欢迎您访问365答案网,请分享给你的朋友!
生活常识 学习资料

Ubuntu下安装CUDA

时间:2023-07-20

注意:
Ubuntu18.04默认GCC-7,由于CUDA 9.x不支持GCC-7(下载页面没有对ubuntu18的支持),所以需要安装低版本的5或者<= 6.3.0,并设置为默认版本。否则应安装10.0以上版本CUDA

查看是否安装了cuda

# 法1cat /usr/local/cuda/version.txt# 法2nvcc --version

若没有安装,则查看是否有N卡驱动,若无N卡驱动,则到软件与更新 -> 附加驱动中安装驱动

查看N卡驱动支持的cuda版本

nvidia-smi

如下图支持最高CUDA版本是11.4

到cuda-toolkit-archive,选择需要的CUDA下载

如下图,选择runfile(local),并使用生成的指令进行下载和安装

若第1步提示Existing package manager installation of the driver found、It is strongly recommended that you remove this before continuing.,选择continue,在下一步中去除driver项,之后选择install:
安装完成后,显示如下:

在~/.bashrc文件中添加如下环境变量:

export PATH=/usr/local/cuda-11.3/bin${PATH:+:${PATH}}export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

验证是否安装成功

nvcc -V

安装cuDNN,到cudnn-archive下载和CUDA对应的版本

1.Navigate to your directory containing the cuDNN tar file.2.Unzip the cuDNN package. $ tar -xvf cudnn-linux-x86_64-8.x.x.x_cudaX.Y-archive.tar.xz3、Copy the following files into the CUDA toolkit directory. $ sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include $ sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 $ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

按照↑教程,可下载cuDNN Library for Linux (x86_64)用复制的方式安装,使用如下命令查看安装版本

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

也可下载Runtime Library和Developer Library的deb包,使用sudo dpkg -i xxx.deb命令进行安装,使用如下命令查看安装版本

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

多版本CUDA安装重复上述步骤即可,最后通过更改第6步中环境变量切换CUDA版本

Copyright © 2016-2020 www.365daan.com All Rights Reserved. 365答案网 版权所有 备案号:

部分内容来自互联网,版权归原作者所有,如有冒犯请联系我们,我们将在三个工作时内妥善处理。