CUDN安装+pytorch

写在前面

cudn版本为11.1. pytorch的版本为1.9(之前1.8改成了1.9,后面对应的版本请看官注意),应该属于比较新的版本

所有下载安装内容我放在百度云盘了,需要的人自取,注意里面还有别的版本别下错了。

链接:https://pan.baidu.com/s/1-_MygT24JvStR_wYwSDpvw 
提取码:kygg

安装CUDA相关

首先需要根据自己的显卡型号以及驱动版本选择对应的CUDA版本。网上有些说根据CUDN安装+pytorch_第1张图片

这里来选择,个人认为还是按照官网的说明来办吧~ 

CUDN安装+pytorch_第2张图片

CUDN安装+pytorch_第3张图片这里我选择的是11.0.3版本的CUDA 

注意由于下载网络质量的原因,建议在安装之前检查一下文件的完整性,对比一下MD5码是否一致:

dabba7135b466d6726eade724d2eb598 cuda_11.0.3_450.51.06_linux_ppc64le.run
70af4cebe30549b9995fb9c57d538214 cuda_11.0.3_450.51.06_linux.run
7f4e95eb69e2241dd278fc2002dfd66a cuda_11.0.3_450.51.06_linux_sbsa.run
80ae0fdbe04759123f3cab81f2aadabd cuda_11.0.3_451.82_win10.exe
1b88bf7bb8e50207bbb53ed2033f93f3 cuda_11.0.3_win10_network.exe
6967f3108245f601d56504bee58a4d11 cuda-repo-cross-sbsa-rhel8-11-0-local-11.0.3-1.noarch.rpm
01628ff2a39a0f002c3e4ade148b12e8 cuda-repo-cross-sbsa-sles15-11-0-local-11.0.3-1.noarch.rpm
d5693bcb4cf72f9f8e2f45634ebc1883 cuda-repo-cross-sbsa-ubuntu1804-11-0-local_11.0.3-1_all.deb
f5ee54dac3c9a1b070830078b928e01b cuda-repo-opensuse15-11-0-local-11.0.3_450.51.06-1.x86_64.rpm
15d9114fe60f10835a14ac0f502e40e2 cuda-repo-rhel7-11-0-local-11.0.3_450.51.06-1.x86_64.rpm
67a6d3ab3551e7b7d78a147d151e384f cuda-repo-rhel8-11-0-local-11.0.3_450.51.06-1.aarch64.rpm
864be984dca2e904713ce1a58a437e8f cuda-repo-rhel8-11-0-local-11.0.3_450.51.06-1.ppc64le.rpm
96c495669d466484e4bd6554d986abf9 cuda-repo-rhel8-11-0-local-11.0.3_450.51.06-1.x86_64.rpm
2bd678bd3207196adeb22ed26c7c14dd cuda-repo-sles15-11-0-local-11.0.3_450.51.06-1.aarch64.rpm
0b44ee3179acef64dc5a22ea6af5af42 cuda-repo-sles15-11-0-local-11.0.3_450.51.06-1.x86_64.rpm
2606b1b7e606af59e7a1555bf55badb5 cuda-repo-ubuntu1604-11-0-local_11.0.3-450.51.06-1_amd64.deb
9ff7717492e88850db704e47c542bbf0 cuda-repo-ubuntu1804-11-0-local_11.0.3-450.51.06-1_amd64.deb
68e7e7e14ce9376a03babdf0e5ca2ff8 cuda-repo-ubuntu1804-11-0-local_11.0.3-450.51.06-1_arm64.deb
e26e734552d0ab3a1541868303e3b644 cuda-repo-ubuntu1804-11-0-local_11.0.3-450.51.06-1_ppc64el.deb
da5bcebd5b01e15025af19ba82c7b40e cuda-repo-ubuntu2004-11-0-local_11.0.3-450.51.06-1_amd64.deb
ea44cd8728806aaf9278cf0929ed032d cuda_11.1.0_455.23.05_linux_ppc64le.run
10c28f59cc2aba9a886d409534b05ddf cuda_11.1.0_455.23.05_linux.run
747c4b8ba50a5bafb0e53d42c64e1865 cuda_11.1.0_455.23.05_linux_sbsa.run
2786aedea8c05d17d21113645d6423e8 cuda_11.1.0_456.43_win10.exe
253029924ebb87c77cf24e8a170c5c82 cuda_11.1.0_win10_network.exe
4f5555f775aee7c85a9c10411a4cc16a cuda-repo-cross-sbsa-rhel8-11-1-local-11.1.0-1.noarch.rpm
96cd7fb41bcdda3ccfbf0e72458765a5 cuda-repo-cross-sbsa-sles15-11-1-local-11.1.0-1.noarch.rpm
d40ebefa3bce113bd94d1735e89df0ec cuda-repo-cross-sbsa-ubuntu1804-11-1-local_11.1.0-1_all.deb
a762a94bbbaf5dbd962b83604877d656 cuda-repo-debian10-11-1-local_11.1.0-455.23.05-1_amd64.deb
a200c6f6d3ca31d1edbd0b73b19c8717 cuda-repo-fedora32-11-1-local-11.1.0_455.23.05-1.x86_64.rpm
ee96a4e7c0c7ac9da74e9b1a23860baf cuda-repo-opensuse15-11-1-local-11.1.0_455.23.05-1.x86_64.rpm
055a18bfa293780208cdab1472b05a90 cuda-repo-rhel7-11-1-local-11.1.0_455.23.05-1.x86_64.rpm
c8953a6be3c69a3c2493fc21cdc16662 cuda-repo-rhel8-11-1-local-11.1.0_455.23.05-1.aarch64.rpm
64bc0f8462c3e6e5f35f255217b8a17c cuda-repo-rhel8-11-1-local-11.1.0_455.23.05-1.ppc64le.rpm
1db27482520d54e9d5d4191413e16bd3 cuda-repo-rhel8-11-1-local-11.1.0_455.23.05-1.x86_64.rpm
e827b21c77e39035597e604d53819dd8 cuda-repo-sles15-11-1-local-11.1.0_455.23.05-1.aarch64.rpm
1fe0a9ed2ba8f060b42178e042bd037d cuda-repo-sles15-11-1-local-11.1.0_455.23.05-1.x86_64.rpm
e5c34a7e08aab0d2a79ff1ea97bb48fd cuda-repo-ubuntu1604-11-1-local_11.1.0-455.23.05-1_amd64.deb
056de5e03444cce506202f50967b0016 cuda-repo-ubuntu1804-11-1-local_11.1.0-455.23.05-1_amd64.deb
2644db04c54065a7f83a8a11caeb0102 cuda-repo-ubuntu1804-11-1-local_11.1.0-455.23.05-1_arm64.deb
88787551c5a9847333d4b66d8387f1c1 cuda-repo-ubuntu1804-11-1-local_11.1.0-455.23.05-1_ppc64el.deb
c4ba145bf9f4b6989293299c5a177ef2 cuda-repo-ubuntu2004-11-1-local_11.1.0-455.23.05-1_amd64.deb
99cdb2dd3d72bc3ae551622a3d615359 cuda-repo-wsl-ubuntu-11-1-local_11.1.0-1_amd64.deb

现在获得了CUDA的本地EXE文件开始安装,双击打开后,会选择一个存放临时文件的文件夹

CUDN安装+pytorch_第4张图片

注意:这里选择一个新建的空文件夹,有可能在安装结束后,自动清空该文件夹目录下的所有文件,这个文件夹在安装结束后会自动删除。

之后安装软件会检测系统环境,注意如果您的电脑C盘空间非常富余,建议选择自定义安装,可以选择文件的安装位置,当然即使这样仍然会有一小部分放在C盘这是在所难免的。(有人说需要取消CUDA目录下VS软件的安装,我没有取消,这个影响待定) ,一路等待结束。。。

在命令行中输入nvcc -V查看安装的版本

CUDN安装+pytorch_第5张图片

安装和CUDN对应的cuDNN

为了保险起见,选择了8.1.0版本的cuDNN

CUDN安装+pytorch_第6张图片

 将下载的文件解压后:

CUDN安装+pytorch_第7张图片

把三个文件夹里的文件还有NVIDIA_SLA_cuDNN_Support.txt文件拷贝到安装CUDA文件夹下:

CUDN安装+pytorch_第8张图片

将CUDA的路径添加到系统环境中:

CUDN安装+pytorch_第9张图片

在环境变量中,确认以下两项:

 CUDN安装+pytorch_第10张图片

 到此CUDA部分结束

pytorch

选择相应的下载组件

官网

CUDN安装+pytorch_第11张图片

CUDN安装+pytorch_第12张图片

这个时候最好不要直接就用这个命令去下载,自己到后面的静态页面下载对应的whl文件,然后本地安装。本地安装,命令窗口到下载的文件路径下,pip install 对应的文件即可~

CUDN安装+pytorch_第13张图片

 CUDN安装+pytorch_第14张图片

还没有结束哦!

测试

import torch 
print(torch.__version__)

CUDN安装+pytorch_第15张图片

看了网上说还需要安装一个包:(这个包因为网络的原因好难下的)

pip install intel-openmp 

但是仍然拯救不了我。。。

不过在github上有人说一顺溜删光就好了,好几个人都说有用,斗胆一试。

我一口气删了三个dll文件,caffe2_detectron_ops.dll、omptarget.rtl.level0.dll、omptarget.rtl.opencl.dll

CUDN安装+pytorch_第16张图片

CUDN安装+pytorch_第17张图片再次运行就成功了,再有什么问题我会再更新的。。。

你可能感兴趣的:(pytorch,cuda)