啪啪欧美精品A片二区

系統城裝機大師 - 唯一官網:www.trattoriacinecitta.com!

當前位置:首頁 > 腳本中心 > python > 詳細頁面

PyTorch兩種安裝方法

時間:2021-03-29來源:www.trattoriacinecitta.com作者:電腦系統城

本文安裝的是pytorch1.4版本(cpu版本)

首先需要安裝Anaconda

是否需要安裝基于cuda的PyTorch版本呢?

對于普通筆記本來說即使有顯卡性能也不高,跑不動層數較深的深度學習網絡,所以就不用裝cuda啦。實際應用時深度學習肯定離不開基于高性能GPU的cuda,作為一般的筆記本,基本都跑不動數據量較大的模型,所以安裝CPU版的PyTorch即可。以后如果繼續進行深度學習的研究或開發,都會基于高性能服務器,此時安裝PyTorch版本肯定是選擇有cuda的版本了。

然后進入PyTorch官網https://pytorch.org/,選擇相應版本的pytorch(本文安裝的是1.4版本的),復制“Run this Command”里面的代碼

在這里插入圖片描述

然后打開Anaconda Prompt

在這里插入圖片描述

運行剛剛在官網復制的命令,如下:

1 conda install pytorch torchvision cpuonly -c pytorch

系統開始進行安裝pytorch

在這里插入圖片描述

等到屏幕上出現Proceed([y]/n)?時,輸入y

在這里插入圖片描述

接下來就開始自動下載安裝了:

在這里插入圖片描述

測試一下是否安裝成功:

首先輸入python,回車。接下來繼續輸入:

1
2
3
4
//測試:
import torch
import torchvision
print(torch.__version__)

在這里插入圖片描述

PyTorch1.4,cpu版本,就安裝成功啦

到此這篇關于PyTorch兩種安裝方法的文章就介紹到這了

分享到:

相關信息

系統教程欄目

欄目熱門教程

人氣教程排行

站長推薦

熱門系統下載