安装CUDA
https://developer.nvidia.com/cuda-toolkit-archive
使用deb安装的话,有时会报错:
dpkg: 处理软件包 nvidia-driver-450 (--configure)时出错:
依赖关系问题 - 仍未被配置
dpkg: 依赖关系问题使得 cuda-drivers-450 的配置工作不能继续:
cuda-drivers-450 依赖于 nvidia-dkms-450 (>= 450.51.05);然而:
软件包 nvidia-dkms-450 尚未配置。
cuda-drivers-450 依赖于 nvidia-driver-450 (>= 450.51.05);然而:
软件包 nvidia-driver-450 尚未配置。
解决办法:
sudo apt clean
sudo apt-get update
sudo apt purge nvidia-*
sudo apt autoremove
最后下载run文件进行安装,把第一项取消,然后install
配置环境
vim ~/.bashrc
export PATH=/usr/local/cuda-11.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH
source ~/.bashrc
确认
$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:09:46_PDT_2020
Cuda compilation tools, release 11.1, V11.1.105
Build cuda_11.1.TC455_06.29190527_0