PyTorch 最新安裝教程
1. 安裝 Anaconda
Anaconda 是一個用於科學計算的 Python 發行版,支持 Linux, Mac, Windows, 包含了眾多流行的科學計算、數據分析的 Python 包。
Anaconda 安裝包可以到 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 下載 or 也可以去官網下載。
2. 檢查顯卡,更新驅動
建議是首先更新驅動,全都按最新的東西來安裝
https://www.nvidia.cn/Download/index.aspx?lang=cn#
在這里選擇自己的顯卡型號,下載安裝即可,下載很快也很簡單,建議大多數人都更新一下顯卡驅動,避免不必要的麻煩。
- 安裝好后進入NVIDIA控制面板,鼠標右鍵可進入

點擊組件找到自己CUDA的版本號

或者 win+R—> cmd ,進入命令行,輸入:
nvidia-smi

如果沒有這條命令,則需要添加環境變量(百度配置)一般電腦都可以直接用,我的顯卡是:GF:MX250
也可獲得版本號:

備注:第二步不一定是必須的,如果你的電腦較新,那么不更新也可以直接進行下面操作,我的電腦是19年買的,中間好像也沒更新驅動,這一步直接跳過,最后也安裝成功了。
3. 創建PyTorch環境
不同的項目需要不同的虛擬環境,可以處理不同版本的項目之間不兼容問題。
1.進入 Anaconda prompt 命令窗口

2. 輸入以下內容:
conda create -n PyTorch python=3.8
PyTorch是虛擬環境名字(可以隨意設置),3.8 是python版本,都可以按自己需求改,一定要指定具體 python 版本。

然后按 y,繼續安裝所需的各種依賴包。
- 創建成功后,輸入以下命令:
conda info --envs
可以看見自己的所有環境

如果出現錯誤,可能是外網下載過慢,需要配置國內鏡像源。
4. 配置清華TUNA鏡像源
TUNA 提供了 Anaconda 倉庫與第三方源(conda-forge、msys2、pytorch等,查看完整列表)的鏡像,各系統都可以通過修改用戶目錄下的 .condarc 文件。Windows 用戶無法直接創建名為 .condarc 的文件,可先執行 conda config --set show_channel_urls yes 生成該文件之后再修改。
注:由於更新過快難以同步,TUNA不同步pytorch-nightly, pytorch-nightly-cpu, ignite-nightly這三個包。
- 生成
.condarc文件
在 Anaconda prompt 命令窗口,中輸入:
conda config --set show_channel_urls yes
之后可以在 C:\Users\xxx 中看到 .condarc 文件

2. 記事本打開 .condarc 文件,重寫其中的內容。
channels: - defaults show_channel_urls: true default_channels: - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2 custom_channels: conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

即可添加 Anaconda Python 免費倉庫。
在Anaconda prompt 命令窗口運行 conda clean -i 清除索引緩存,保證用的是鏡像站提供的索引。
5. 安裝 PyTorch
-
進入pytorch官網:https://pytorch.org/
-
網頁下拉,即可看到下圖,官網會自動根據你的電腦,顯示的即是你可安裝的CUDA版本,並給出安裝命令。

- 復制官網提供的命令
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
- 打開
Anaconda prompt命令窗口,進入你剛剛所創建的環境(我的命名是PyTorch)
conda activate PyTorch
進入環境

5. 最后輸入官網提供的命令,即可下載
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch


每個人的命令,會由於電腦配置而各不同
注意:下載安裝過程,可能因為某些原因,並不是很順利,但遇到問題不要慌,另外網速一定要好,避免出現不必要的錯誤。
6. 測試
打開 Anaconda prompt 命令窗口,激活環境,輸入python,進入python開發環境中
import torch
torch.cuda.is_available()
True

看到True的那一刻,我真的開心,終於成功了。
