Ubuntu系統Anaconda安裝Pytorch,教你如何優雅的安裝環境(1-5)


一、明確前提

首先我們安裝的PyTorch是cuda版本的,所以首先要知道你的cuda版本(我的cuda是 10.2)

二、打開pytorch官網,查看安裝命令

https://pytorch.org/get-started/locally/

 

這里我沒有安裝最新版本怕不穩定

 

 

 選擇一個適合你的版本(先別急輸入命令哦!)

a)官網安裝

conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.2 -c pytorch

b)鏡像安裝(去掉 -c

conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.2 pytorch

確定你已經配置過pytorch的國內鏡像,如果使用國內鏡像安裝不成功就多試兩次,總是不行可以選擇官網安裝(親測!有時官網安裝也不慢哦!)

 

 

 

 

三、創建新的anaconda環境

conda create -n python36torch15 python=3.6

環境創建好之后,輸入之前找到的安裝pytorch的命令

 

四、驗證安裝

那你的cuda版本的pytorch安裝好了沒有呢?見證奇跡的時刻到了。

在你創建的anaconda環境中(我的是python36torch15)輸入

python

進入python代碼編寫環境

輸入代碼

#判斷是否安裝了cuda
import torch
print(torch.cuda.is_available())  #返回True則說明已經安裝了cuda

——————  到這里pytorch就裝好了開始你的pytorch之旅吧  ——————

 

 

 

******  下面這個是贈送的福利哦!相信你后面會用到的  ******

#判斷是否安裝了cuDNN
from torch.backends import  cudnn 
print(cudnn.is_available())  #返回True則說明已經安裝了cuDNN

 

最后~祝科研順利啊!!!


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM