手把手教你在Windows10環(huán)境下安裝深度學習框架（pytorch or tensorflow）

• 1. 安裝Anaconda：（常用的python版本和各類包管理器）
• • 1.1. 下載地址：
  • 1.2. 配置環(huán)境變量：
  • 1.3. 檢查配置情況：
  • 1.4. 切換anaconda下載文件的源：
• 2. 安裝CUDA：（顯卡廠商NVIDIA推出的并行運算平臺）
• 3. 安裝CUDNN：（用于深度神經網絡的GPU加速庫）
• 4. 安裝pytorch（GPU版本）
• 5. 安裝tensorflow（GPU版本）
• 補充：

1. 安裝Anaconda：（常用的python版本和各類包管理器）

1.1. 下載地址：

https://www.anaconda.com/products/individual
不一定安裝在c盤下，自行選擇

1.2. 配置環(huán)境變量：

環(huán)境變量配置路徑：此電腦右鍵-屬性-高級系統(tǒng)設置-環(huán)境變量（根據(jù)自己安裝anaconda目錄名稱輸入），在用戶變量或者系統(tǒng)變量中輸入均可

C:\Users\yuan\anaconda3 C:\Users\yuan\anaconda3\Scripts C:\Users\yuan\anaconda3\Library\bin C:\Users\yuan\anaconda3\Library\usr\bin C:\Users\yuan\anaconda3\Library\mingw-w64\bin

1.3. 檢查配置情況：

cmd命令行中輸入 conda --version（注：- - 查看輸出情況是否為版本號）

1.4. 切換anaconda下載文件的源：

默認外國網站，配完鏡像源，使用國內的網站，下載速度會提升

命令如下：配置清華源

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/conda-forge/

設置搜索時顯示通道地址

conda config --set show_channel_urls yes

顯示源

conda config --show channels

刪除源

conda config --remove channels 源名稱或鏈接

2. 安裝CUDA：（顯卡廠商NVIDIA推出的并行運算平臺）

安裝之前輸入nvidia-smi查看自己的顯卡驅動版本，找到對應的cuda版本。30系顯卡建議直接cuda11.1以上版本。

Cuda下載地址：https://developer.nvidia.com/cuda-toolkit-archive
直接打開安裝exe安裝，建議安裝在默認路徑下，測試安裝情況命令行命令：nvcc -V (注：大寫的V)

3. 安裝CUDNN：（用于深度神經網絡的GPU加速庫）

下載地址：https://developer.nvidia.com/cudnn
需要先注冊賬號，看自己的cuda版本下載對應的cudnn版本
Cuda11以上版本對應cudnn11.3、cuda10版本對應cudnn10
下載完成之后

1) 將cuda\bin目錄中的 cudnn64_8.dll 復制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\bin 中 2) 將cuda\include目錄中的 cudnn.h 復制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\include中 3) 將cuda\lib\x64 目錄中的 cudnn.lib 復制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\lib\x64 中 4) 然后將C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2和 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\lib\x64都添加環(huán)境變量path中。

4. 安裝pytorch（GPU版本）

前往：https://pytorch.org/get-started/locally/ 查看安裝命令，有條件可以選擇直接用conda或者pip命令安裝，國內的選擇在pip命令中輸入清華源，命令如下：

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch===1.5.1 torchvision===0.6.1 -f https://download.pytorch.org/whl/torch_stable.html

5. 安裝tensorflow（GPU版本）

conda install tensorflow-gpu==1.4.0 或者 pip install tensorflow-gpu==1.4.0

補充：

為方便拓展建議在anaconda內新建開發(fā)環(huán)境，方便之后進一步的配置，anaconda可以直接建立特定python版本的環(huán)境。
安裝完anaconda之后
創(chuàng)建環(huán)境：

conda create --name env_name python=3.7 #env_name(命名你的環(huán)境，英文) #python=3.7（代表3.7版本的python，可以修改為其他）

創(chuàng)建環(huán)境之后激活環(huán)境：activate env_name 代表你進入了該環(huán)境目錄下，可以調用該環(huán)境下所有的包
安裝框架和常用包：

Conda install tensorflow-gpu==1.4.0 Conda install torch Conda install numpy Conda install ..........

其他命令：

刪除環(huán)境：conda remove -n env_name --all 推出環(huán)境：conda deactivate env_name 查看所有環(huán)境：conda info --envs 查看該環(huán)境目錄下所有的包：conda list 安裝包：conda install pkg_name 卸載包：conda uninstall pkg_name

總結

以上是生活随笔為你收集整理的手把手教你在Windows10环境下安装深度学习框架（pytorch or tensorflow）的全部內容，希望文章能夠幫你解決所遇到的問題。

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