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/