Ubuntu20.04安装cuda10.1的步骤(图⽂教程)
安装前准备
CUDA的主要⽤途是深度学习,⽽⽬前主流的深度学习框架Tensorflow2最⾼⽀持CUDA 10.1,因此本⽂讲解在Ubuntu 20.04系统上安装CUDA 10.1的主要过程。
⾸先要查看你的NVIDIA显卡驱动是否⽀持cuda10.1版本
在终端执⾏以下命令:
nvidia-smi
如果 CUDA Version: … 这⾥的版本号⼤于等于10.1(我的是10.2),就可以安装cuda10.1.
关键点:gcc降级
因为Ubuntu20.04⾃带的gcc版本为9.3,⽽cuda10.1不⽀持gcc-9,因此要⼿动安装gcc-7,命令如下:
sudo apt-get install gcc-7 g++-7
安装完gcc-7,系统中就存在两个版本的gcc,因此要设置默认的gcc,命令如下:
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 9
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 1
此命令可以通过update-alternatives设置gcc各版本的优先级,优先级最⾼的为系统默认版本,可以⽤下述命令显⽰其优先级:
sudo update-alternatives --display gcc
设置默认的g++也是如此:
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-7 9
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-9 1
显⽰g++优先级:
sudo update-alternatives --display g++
按照官⽹的安装步骤下载并安装CUDA 10.1
如上图所⽰,推荐选择run⽂件安装,⽐较简洁。
当安装界⾯启动,选择continue。
然后输⼊accept,回车。
由于你的系统中已经有了NVIDIA显卡驱动,如果不想安装CUDA 10.1中附带的驱动,移动到Driver选项上,按空格键将该项取消。如下图。
移动到Install选项,回车,等待安装完成
配置可执⾏⽂件及库路径
安装完成后,需要为CUDA 10.1配置环境。
⾸先打开家⽬录下的.bashrc⽂件:gedit ~/.bashrc
在⽂件的末尾添加如下内容:
export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}ubuntu安装教程
保存并关闭⽂件,完成配置。
更新环境:source ~/.bashrc
安装cuDNN
下载cuDNN
打开链接后,⽹站会要求你登陆,如果没有NVIDIA账号,注册后再登陆。
选择与CUDA 10.1对应的版本(7.6.5),点开后选择,点击下载。(最好选择 cuDNN Library for Linux 这个⽂件格式安装⽐较⽅便)
安装cuDNN
切换到下载⽂件所在⽬录,解压下载好的cuDNN压缩⽂件到当前⽬录:
tar zxvf ./cudnn-10.1-linux-x64-v7.6. -C ./
将解压出的
cuda/include/cudnn.h⽂件复制到/usr/local/cuda/include⽂件夹
cuda/lib64/下所有⽂件复制到/usr/local/cuda/lib64⽂件夹
为上述⽂件添加读取和执⾏权限:
sudo chmod 755 /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
安装完毕
参考⽂章:
到此这篇关于Ubuntu20.04安装cuda10.1的步骤(图⽂教程)的⽂章就介绍到这了,更多相关Ubuntu20.04安装cuda10.1内容请搜索以前的⽂章或继续浏览下⾯的相关⽂章希望⼤家以后多多⽀持!