pytorch的cuda環境搭建（GPU版本安裝）

第一步：安裝cuda

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

第二步：安裝cudnn

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

打開壓縮包，依次將壓縮包中的3個文件復制到相應位置

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

到官網下載對應GPU版本的pytorch

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

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

參考

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