第一步去网址下载
CUDA Toolkit Archive | NVIDIA Developer
然后选择版本,进入网址选择所需版本
选择版本之类的
下拉到下边,俩条命令,第一条下载,第二条安装
第一条命令下载,这种就等等,网好就快,网不好就慢
下载完了,然后运行第二条命令
到这选择accept
这里能全选就全选,假设出现报错,如图所示
出现上边这个报错,就把第一个第二个取消勾选就能安装成功,这块不太懂,有懂得评论区告我一下谢谢
这步就安装成功了,然后就是添加环境变量
命令:
sudo vim ~/.bashrc
添加环境变量
export CUDA_HOME=/usr/local/cuda-11.4
export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-11.4/bin:$PATH
添加完退出
退出命令:点击完Esc 输入 :wq
:wq
出来之后立马输入source ./bashrc
命令:
source ~/.bashrc
输入nvcc -V验证一下,如下就没啥问题了
命令:
nvcc -V
然后输入几条命令cudnn也会自动搞上,第二条有用,另外俩条个人需要
sudo apt install cuda-toolkit-11-4 -y
sudo apt install libcudnn8-dev -y
sudo apt install tensorrt -y