100字范文,内容丰富有趣,生活中的好帮手!
100字范文 > Ubuntu20.04+GTX1060+显卡驱动+CUDA11.8+cuDNN8.5.0

Ubuntu20.04+GTX1060+显卡驱动+CUDA11.8+cuDNN8.5.0

时间:2020-12-16 04:51:19

相关推荐

Ubuntu20.04+GTX1060+显卡驱动+CUDA11.8+cuDNN8.5.0

注意: 不要采用之前安装在UBUNT18.04时 所用的sudo ubuntu-drivers autoinstall 命令自动安装驱动,安装后了造成内核崩溃, WIFI 适配器也无法识别!

安装CUDA 三件套:

1.NVIDIA GPU 驱动 (我的是GTX 1060)

2.Cuda toolkit 11.2

3.cuDNN (给DNN用的驱动)

安装驱动, 参考链接: /datacenter/tesla/tesla-installation-notes/index.html#ubuntu-lts

$ sudo apt-get install linux-headers-$(uname -r)$ distribution=$(. /etc/os-release;echo $ID$VERSION_ID | sed -e 's/\.//g')$ wget https://developer./compute/cuda/repos/$distribution/x86_64/cuda-keyring_1.0-1_all.deb$ sudo dpkg -i cuda-keyring_1.0-1_all.deb$ sudo apt-get update$ sudo apt-get -y install cuda-drivers

安装后检查:

$ nvidia-smi

2. 安装CUDA Toolkit 11.2, 采用方法是 install CUDA toolkit from CUDA repository。

先点击这里的官网,下载

wget https://developer./compute/cuda/repos/ubuntu/x86_64/cuda-ubuntu.pinsudo mv cuda-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600wget https://developer./compute/cuda/11.2.0/local_installers/cuda-repo-ubuntu-11-2-local_11.2.0-460.27.04-1_amd64.debsudo dpkg -i cuda-repo-ubuntu-11-2-local_11.2.0-460.27.04-1_amd64.debsudo apt-key add /var/cuda-repo-ubuntu-11-2-local/7fa2af80.pubsudo apt-get updatesudo apt-get -y install cuda

使用命令

$ nvcc -V

检查是否安装成功, 如果没成功, 用以下命令添加环境变量即可:

echo 'export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}' >> ~/.bashrcsource .bashrc

以下是示列:

note: 估计官方的命令错误, 安装时命令显示的是cuda 11.2版本, 装好后成了11.8的版本。

安装cuDNN

官网下载 : 下载时需要NVIDIA账户, 没有的话注册一个即可。

Download cuDNN v8.5.0 (August 8th, ), for CUDA 11.x 下的

Local Installer for Ubuntu20.04 x86_64 (Deb)

下载后点击运行即可。

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。