ubuntu安裝cuda10.0

前提是先安裝好驅動程序，驅動程序是根據顯卡的型號來選擇的

1、安裝CUDA Toolkit

cuda10.0地址：https://link.zhihu.com/?target=https%3A//developer.nvidia.com/cuda-10.0-download-archive

2、查看ubuntu版本，然后下載對應的CUDA

lsb_release -a

3、復制文件的下載地址，ubuntu終端輸入命令下載

wget https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_410.48_linux

添加可執行權限

chmod a+x ./cuda_10.0.130_410.48_linux

以管理員權限運行

sudo ./cuda_9.0.176_384.81_linux.run

除了第一步的驅動程序不要安裝，選擇n ，其他的都選擇 y

4、添加到環境變量

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

5、使得配置生效

source ~/.bashrc

6、查看

nvcc -V

tensorflow2查看是否可用GPU

tf.test.is_gpu_available()

pytorch查看

torch.cuda.is_available()

如果是tensorflow2 ，似乎還要安裝cuDNN，（https://developer.nvidia.com/rdp/cudnn-archive）

tar -zxvf cudnn-10.0-linux-x64-v7.5.1.10.tgz

cd cuda

sudo cp lib64/lib* /usr/local/cuda-10.0/lib64/

sudo cp include/cudnn.h /usr/local/cuda-10.0/include/