使用Conda配置PyTorch

1. 添加channels

下載地址

$ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ $ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ $ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/ $ conda config --set show_channel_urls yes

2. 創(chuàng)建一個(gè)新環(huán)境

創(chuàng)建一個(gè)新python環(huán)境,為pytorch提供資源.

win10下找到 anaconda的提示行,輸入如下:

(base) C:\Users\54328>conda create -n pytorch python=3.6

• -n: 表示 name 代表,后面接的環(huán)境的名稱
• 注意: 一定國(guó)內(nèi)一定要添加channels, 下載速度會(huì)變得很快~ 而且不容易被墻

等待一會(huì)兒…

[外鏈圖片轉(zhuǎn)存失敗,源站可能有防盜鏈機(jī)制,建議將圖片保存下來(lái)直接上傳(img-678fKdz6-1588215073554)(C:\Users\54328\AppData\Roaming\Typora\typora-user-images\image-20200430003733154.png)]

安裝完成~

conda env list

上面安裝好了python 3.6的環(huán)境,并命名pytorch. 下面激活該環(huán)境

conda activate pytorch

當(dāng)在小黑框里面看見(jiàn)(pytorch) C:\Users\54328的字樣時(shí)證明安裝成功。

下面查看當(dāng)前環(huán)境下由哪些包

pip list

3. 安裝pytorch

pytorch - 官網(wǎng).

進(jìn)入之后選擇:

• PyTorch版本
• 操作系統(tǒng)
• 使用什么安裝(windows下推薦conda)
• 語(yǔ)言(這里選擇Python)
• CUDA: 這個(gè)是顯卡相關(guān)的.
  • 查看GPU是否支持CUDA
  • 在上面如果找到了對(duì)應(yīng)的型號(hào): 那么勾選 CUDA 9.2
  • 如果沒(méi)有找到,勾選 None

完成以上操作之后,你會(huì)得到一串安裝PyTorch的指令.

之后你就打開(kāi)Anaconda Prompt(win10aconda Prompt(win10在左下角的菜單里面找), 輸入(你的也許和我不同):

conda install pytorch torchvision cudatoolkit=10.1

注意,官網(wǎng)會(huì)提供一個(gè) -c pytorch意思是使用 PyTorch官網(wǎng)提供的channel. 這樣在或內(nèi)會(huì)下載很慢.因此 需要去掉

如果上面安裝比較慢, 可以參考這篇文章.之后再嘗試以上代碼

4. 判斷環(huán)境是否安裝成果

[檢驗(yàn)PyTorch是否安裝成功]

• 命令行輸入 python, 進(jìn)入python環(huán)境.
• 輸入import torch, 未報(bào)錯(cuò)則安裝成功

[下面檢驗(yàn)是否能使用GPU]

• 命令行輸入torch.cuda.is_available(), 返回True則代表可以使用GPU

創(chuàng)作挑戰(zhàn)賽新人創(chuàng)作獎(jiǎng)勵(lì)來(lái)咯，堅(jiān)持創(chuàng)作打卡瓜分現(xiàn)金大獎(jiǎng)

總結(jié)

以上是生活随笔為你收集整理的python --- 使用conda配置pytorch的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。