今天在服务器上跑代码,发现报错,说是CUDA版本不对,然后看了一下服务器的版本,发现是9.0,这就有问题了啊,3090的显卡得用11.0上的版本啊,所以接着配置一下深度学的环境,记录一下方便以后查阅。
Ubuntu18.04安装CUDA11.3和cuDNN8.2.0
CUDA11.3
官网地址:https://developer.nvidia.com/cuda-toolkit-archive
下载自己想要的版本:
根据自己电脑的配置选择相应的选项:
wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux.run
找到下载文件,赋予运行权限
sudo chmod a+x cuda_11.3.0_465.19.01_linux.run
运行安装文件
sudo ./cuda_11.3.0_465.19.01_linux.run
输入:accept
因为我们之前已经安装好显卡驱动了,所以这里我就不选择安装:
设置环境变量
sudo vim ~/.bashrc
export PATH=/usr/local/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH
保存文件,让环境变量生效
source ~/.bashrc
验证CUDA Toolkit
nvcc -V
cuDNN8.2.0
官网地址:https://developer.nvidia.com/rdp/cudnn-archive
下载下面红框中的四个文件:
依次执行一下命令:
tar -zxvf cudnn-11.3-linux-x64-v8.2.0.53.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
sudo dpkg -i libcudnn8_8.2.0.53-1+cuda11.3_amd64.deb
sudo dpkg -i libcudnn8-dev_8.2.0.53-1+cuda11.3_amd64.deb
sudo dpkg -i libcudnn8-samples_8.2.0.53-1+cuda11.3_amd64.deb
拷贝例程到用户目录
cp -r /usr/src/cudnn_samples_v8/ $HOME
转到minist例程
cd $HOME/cudnn_samples_v8/mnistCUDNN
编译例程
sudo make clean && make
编译mnistCUDNN时出错:fatal error: FreeImage.h: No such file or directory
sudo apt-get install libfreeimage3 libfreeimage-dev
重新编译后,运行
sudo make clean && make
./mnistCUDNN
如果cudnn被恰当的安装,将输出打印消息:Test passed!
cudnn 检测版本号
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
到此CUDA11.3和cuDNN8.2.0安装成功!!!