安裝cuda和cudnn

 

原來機器上有安裝一個cuda 10.0，想要再安裝一個cuda 10.1

安裝CUDA

1.首先下載.run文件

https://developer.nvidia.com/cuda-10.1-download-archive-base?target_os=Linux&target_arch=x86_64&target_distro=CentOS&target_version=7&target_type=runfilelocal

 

 

 

2.首先給.run文件執行權限

chmod a+x cuda_10.1.105_418.39_linux.run

3.運行

sh cuda_10.1.105_418.39_linux.run

1）敲accept

 2）去掉驅動下載，因為這里我的驅動已經下好

 3）yes

 4） 

 

 

 需要進行下設置

1》

(base) [root@10_1_5_103 var]# bash -c "echo /usr/local/cuda-10.1/lib64/ > /etc/ld.so.conf.d/cuda.conf"
(base) [root@10_1_5_103 var]# ldconfig

 2》添加到~/.bashrc：

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64 
export PATH=$PATH:/usr/local/cuda-10.1/bin

export CUDA_HOME=/usr/local/cuda-10.1 

 然后運行source ~/.bashrc

3》同時更改：

# vim /etc/profile
# source /etc/profile

加入：

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export PATH=$PATH:/usr/local/cuda-10.1/bin
export CUDA_HOME=/usr/local/cuda-10.1 

4》查看cuda版本：

(base) [root@... var]# nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Fri_Feb__8_19:08:17_PST_2019
Cuda compilation tools, release 10.1, V10.1.105

 

4.測試例子效果：

cd .../NVIDIA_CUDA-10.1_Samples/1_Utilities/bandwidthTest
make
./bandwidthTest

返回：

 

5.查看torch調用的cuda：

>>> import torch
>>> print(torch.version.cuda) 10.0

但是這不說明其使用的真的是這個，查看：

>>> import torch.utils.cpp_extension
>>> torch.utils.cpp_extension.CUDA_HOME '/usr/local/cuda-10.1'

可見改了

 

安裝cuDNN

打開鏈接https://developer.nvidia.com/cudnn，需要先注冊和填寫信息，之后選擇對應版本的安裝包

 

1.版本8

1）解壓：

tar -zxvf cudnn-10.1-linux-x64-v8.0.5.39.tgz

會在當前文件下生成一個cuda文件夾

2）操作：

sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include/ 
sudo cp cuda/include/cudnn_version.h /usr/local/cuda-10.1/include/ 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64/ 
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h 
sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*

3）查看版本：

cat /usr/local/cuda-10.1/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

返回：

#define CUDNN_MAJOR 8
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 5
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#endif /* CUDNN_VERSION_H */

 

2.版本7

如果是cudnn7版本的：

1）解壓：

tar -zxvf cudnn-10.1-linux-x64-v7.6.5.32.tgz

會在當前文件下生成一個cuda文件夾

2）操作：

sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include/ 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64/ 
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h 
sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*

3）查看版本：

cat /usr/local/cuda-10.1/include/cudnn.h | grep CUDNN_MAJOR -A 2

返回：

#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 5
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"