文章目錄

• 1. 配置conda虛擬環境
• 2. 安裝Pytorch
• • 2.1 官網安裝
  • 2.2 清華鏡像安裝
  • 2.3 anaconda網站自選安裝
• 3. 驗證
• 參考資料

1. 配置conda虛擬環境

（1）打開Anaconda Prompt

（2）輸入命令

conda create -n pytorch python = 3.6

接著輸入y，便可完成pytorch虛擬環境創建。

（3）進入Pytorch虛擬環境
輸入：conda activate pytorch 進入到pytorch環境

如果能夠順利進入，則證明第一步安裝成功。

2. 安裝Pytorch

2.1 官網安裝

（1）打開pytorch官網：https://pytorch.org/

上述配置主要需要注意的是系統、Pytorch版本。安裝時，將最后一行的Command代碼復制到conda命令行窗口運行即可

（2）Pytorch版本選擇及注意事項
CUDA
如果選擇安裝CUDA版本需要注意：

電腦是否有英偉達（NVIDIA）的顯卡。可以通過“右鍵 --> 任務管理器 --> 性能 --> GPU”查看。如果沒有NVIDIA顯卡，則可以選擇CPU版本安裝。

如果有NVIDIA顯卡，接下來需要檢查是否安裝CUDA.

如果沒有安裝cuda，則可以進入官網：https://developer.nvidia.com/zh-cn/cuda-toolkit 下載。等待下載完安裝即可。

檢查驅動版本是否大于396.2：輸入 nvidia-smi 查看驅動信息

如果版本數小于要求，則可以進入英偉達驅動官網https://www.nvidia.com/Download/index.aspx?lang=cn
選擇適合自己電腦的驅動程序然后下載更新

2.2 清華鏡像安裝

（3）清華鏡像安裝
如果直接按照命令行代碼：conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch 運行，可能下載的速度會很慢。可以嘗試以下方法更換成清華鏡像下載
運行以下命令：

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 --set show_channel_urls yes conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64

再輸入conda config --show channels檢查已經加入的channels

最后將命令復制到命令行，刪除 -c pytorch后增加地址https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64

如果上述方法仍然無法安裝，則可嘗試（4）

（4）用pip加清華源下載
輸入命令 pip install torch -i https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ 即可完成安裝

2.3 anaconda網站自選安裝

進入官網：https://anaconda.org

在搜索框中輸入需要安裝的庫，如pytorch

選擇其中一個鏡像，按照命令安裝即可。

采用上述方法安裝也是很快的，整個安裝過程不超過5分鐘。（在安裝pytorch時可能會有一段時間的“卡頓，不要放棄，稍等一會兒就好”。）

3. 驗證

（1）pytorch安裝驗證

能夠如上所示正常運行，表示pytorch安裝成功。

（2）torch.cuda.is_available()驗證
輸入 torch.cuda.is_available() ，如果返回true則表示GPU可以使用。

參考資料

[1] https://www.bilibili.com/video/BV1hE411t7RN?p=1
[2] https://www.bilibili.com/video/BV1ES4y1L7BY?p=3&spm_id_from=pageDriver
[3] https://blog.csdn.net/zzq060143/article/details/88042075

總結

以上是生活随笔為你收集整理的Pytorch安装指南（cuda、镜像下载）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。