标签：usr 3.7 python root 7a927d2dc743 CUDNN CUDA 安装 cuda

安装CUDA、CUDNN

可以去英伟达官网下载、安装最新的的版本，也可以根据自己需求去CUDA历史版本、CUDNN历史版本下载自己需要的版本。附送CUDA、CUDNN与tensorflow版本对应关系。

1、安装CUDA

1.1 下载cuda

(python-3.7) root@7a927d2dc743:~# wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run

(python-3.7) root@7a927d2dc743:~# ll -d cuda_10.2.89_440.33.01_linux.run 
-rw-r--r-- 1 root root 2645419389 Nov 13  2019 cuda_10.2.89_440.33.01_linux.run

1.2 安装cuda

(python-3.7) root@7a927d2dc743:~# sh cuda_10.2.89_440.33.01_linux.run --silent --toolkit --samples --librarypath=/usr/local/cuda-10.2

1.3 检查安装的版本

(python-3.7) root@7a927d2dc743:~# ln -s /usr/local/cuda/bin/nvcc /usr/bin/nvcc-python-3.7
(python-3.7) root@7a927d2dc743:~# which nvcc-python-3.7
/usr/bin/nvcc-python-3.7

(python-3.7) root@7a927d2dc743:~# nvcc-python-3.7 -V
nvcc-python-3: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:24:38_PDT_2019
Cuda compilation tools, release 10.2, V10.2.89

1.4 添加环境变量

#添加环境变量到 ~/.bashrc 文件的末尾
(python-3.7) root@7a927d2dc743:~# vim ~/.bashrc
(python-3.7) root@7a927d2dc743:~# tail -3 ~/.bashrc
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

#激活环境变量
(python-3.7) root@7a927d2dc743:~# source ~/.bashrc 

1.5 测试 CUDA Toolkit 以验证是否安装成功

#Result = PASS则安装成功
(python-3.7) root@7a927d2dc743:~# cd /usr/local/cuda/extras/demo_suite/
(python-3.7) root@7a927d2dc743:/usr/local/cuda/extras/demo_suite# ./deviceQuery 
......
deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 11.6, CUDA Runtime Version = 10.2, NumDevs = 4, Device0 = NVIDIA GeForce RTX 2080 Ti, Device1 = NVIDIA GeForce RTX 2080 Ti, Device2 = NVIDIA GeForce RTX 2080 Ti, Device3 = NVIDIA GeForce RTX 2080 Ti
Result = PASS

2、安装CUDNN

2.1 下载cudnn

(python-3.7) root@7a927d2dc743:~# ll -d cudnn-10.2-linux-x64-v7.6.5.32.tgz 
-rw-r--r-- 1 root root 548210361 Mar 29 09:54 cudnn-10.2-linux-x64-v7.6.5.32.tgz

#解压缩
(python-3.7) root@7a927d2dc743:~# tar xf cudnn-10.2-linux-x64-v7.6.5.32.tgz
(python-3.7) root@7a927d2dc743:~# ll -d cuda
drwxr-xr-x 4 root root 70 Mar 29 09:55 cuda/

2.2 安装cudnn

#把相应的文件，复制到指定目录即可
(python-3.7) root@7a927d2dc743:~# cp cuda/include/cudnn* /usr/local/cuda/include/ 
(python-3.7) root@7a927d2dc743:~# cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/

#添加权限
(python-3.7) root@7a927d2dc743:~# chmod a+r /usr/local/cuda/include/cudnn* /usr/local/cuda/lib64/libcudnn*

标签：usr,3.7,python,root,7a927d2dc743,CUDNN,CUDA,安装,cuda
来源： https://www.cnblogs.com/ywb123/p/16075391.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。