使用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. 創建一個新環境

創建一個新python環境,為pytorch提供資源.

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

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

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

等待一會兒…

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

安裝完成~

conda env list

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

conda activate pytorch

當在小黑框里面看見(pytorch) C:\Users\54328的字樣時證明安裝成功。

下面查看當前環境下由哪些包

pip list

3. 安裝pytorch

pytorch - 官網.

進入之后選擇:

• PyTorch版本
• 操作系統
• 使用什么安裝(windows下推薦conda)
• 語言(這里選擇Python)
• CUDA: 這個是顯卡相關的.
  • 查看GPU是否支持CUDA
  • 在上面如果找到了對應的型號: 那么勾選 CUDA 9.2
  • 如果沒有找到,勾選 None

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

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

conda install pytorch torchvision cudatoolkit=10.1

注意,官網會提供一個 -c pytorch意思是使用 PyTorch官網提供的channel. 這樣在或內會下載很慢.因此 需要去掉

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

4. 判斷環境是否安裝成果

[檢驗PyTorch是否安裝成功]

• 命令行輸入 python, 進入python環境.
• 輸入import torch, 未報錯則安裝成功

[下面檢驗是否能使用GPU]

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