20.04两天前发布了,但是CUDA for 20.04什么时候会发布呢?Nvidia通常会迅速跟进吗?
截至2020年6月16日,我成功在Ubuntu 20.04上安装了CUDA 10.1和cuDNN 7.6.5,并且它们与Tensorflow 2.2.0完美兼容。
以下是我遵循的步骤:
1- 安装CUDA(10.1):
目前针对Ubuntu 20.04没有deb文件或运行文件,所以唯一的解决方案是执行以下命令:
sudo apt install nvidia-cuda-toolkit
安装需要一些时间。
之后为了确保CUDA已经安装成功,请运行以下命令:
nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243
whereis cuda
2- 安装 cuDNN (7.6.5):
首先访问此 链接,然后选择 下载 cuDNN。您将被要求登录/创建一个帐户。登录后,您将被要求接受 cuDNN 软件许可协议的条款。
将显示可下载的 cuDNN 列表,请点击 下载 cuDNN v7.6.5 (2019年11月5日),适用于 CUDA 10.1,然后选择 Linux 平台的 cuDNN 库。
下载完成后,解压文件,然后打开终端并运行:
cd cudnn-10.1-linux-x64-v7.6.5.32 # or whatever folder you got after extracting the file
然后:
sudo cp cuda/include/cudnn.h /usr/lib/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/lib/cuda/lib64/
sudo chmod a+r /usr/lib/cuda/include/cudnn.h /usr/lib/cuda/lib64/libcudnn*
~/.bashrc
文件中。您需要运行:echo 'export LD_LIBRARY_PATH=/usr/lib/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/lib/cuda/include:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
import tensorflow as tf
tf.config.list_physical_devices('GPU')
Ubuntu 20.04 LTS和CUDA 11.1安装设置:
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
sudo apt update
sudo apt install -y nvidia-kernel-source-455
sudo apt install -y nvidia-dkms-455
sudo apt install -y nvidia-driver-455
sudo apt install -y cuda-drivers-455
sudo apt install -y cuda-runtime-11-1
sudo apt install -y cuda-11-1
echo 'export PATH=/usr/local/cuda-11.1/bin${PATH:+:${PATH}}' >> $HOME/.bashrc
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda-repo-ubuntu2004-11-0-local_11.0.2-450.51.05-1_amd64.deb
sudo apt install ./cuda-repo-ubuntu2004-11-0-local_11.0.2-450.51.05-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-0-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
在Ubuntu 20.04中已经内置了CUDA包,您可以通过输入以下命令进行安装:
sudo apt-get install nvidia-cuda-toolkit
nvidia-cuda-toolkit
没有起到任何帮助作用。 - paulplusx