原来机器上有安装一个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"
Original: https://www.cnblogs.com/wanghui-garcia/p/14973930.html
Author: 慢行厚积
Title: 安装cuda和cudnn
原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/553824/
转载文章受原作者版权保护。转载请注明原作者出处!