安装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

安装cuda和cudnn

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

安装cuda和cudnn

2)去掉驱动下载,因为这里我的驱动已经下好

安装cuda和cudnn

3)yes

安装cuda和cudnn

4)

安装cuda和cudnn

需要进行下设置

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

返回:

安装cuda和cudnn

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,需要先注册和填写信息,之后选择对应版本的安装包

安装cuda和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"

Original: https://www.cnblogs.com/wanghui-garcia/p/14973930.html
Author: 慢行厚积
Title: 安装cuda和cudnn

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/553824/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球