第一步：安裝cuda

安裝cuda，這里還需要安裝對應(yīng)版本的Visual Studio（具體參考本博主的博客 https://blog.csdn.net/qq_36653505/article/details/81368346 ）

第二步：安裝cudnn

cudnn是pytorch搭建深度學(xué)習(xí)模型的依賴，沒有它，不能運行卷積等操作。
下載對應(yīng)版本的cudnn鏈接：https://developer./rdp/cudnn-archive

打開壓縮包，依次將壓縮包中的3個文件復(fù)制到相應(yīng)位置

1. C:\cuda\bin\cudnn64_7.dll —> C:\Program Files\NVIDIA GPUComputing
   Toolkit\CUDA\v9.1\bin

2. C:\cuda\include\cudnn.h —> C:\Program Files\NVIDIA GPUComputing
   Toolkit\CUDA\v9.1\include

3. C:\cuda\lib\x64\cudnn.lib —> C:\Program Files\NVIDIA GPUComputing
   Toolkit\CUDA\v9.1\lib\x64

注意： 一定要安裝cudnn，不然無法進行卷積操作

安裝GPU版本的pytorch

到官網(wǎng)下載對應(yīng)GPU版本的pytorch

然后使用pip install 文件名安裝下載好的pytorch壓縮包

第四步：檢查是否安裝成功

參考

1. https://blog.csdn.net/MarsLee_U/article/details/80489618
2. https://blog.csdn.net/u011622208/article/details/80275344