要先装显卡驱动
进入官网:CUDA Toolkit 11.3 Downloads | NVIDIA Developer
或通过下面方式进
官网界面如下:
用下图的方式下载 ,CUDA Toolkit 11.3 Downloads | NVIDIA Developer
选定网址后右键选择第三个转入下载cuda11.3包
下载完成后,在包含cuda_11.3.0_465.19.01_linux.run的文件夹中打开终端
输入如下命令:
sudo bash cuda_11.3.0_465.19.01_linux.run
依次选定 Continue accept
选择安装前,去掉第一个Driver
添加自己需要的cuda版本到环境变量, vim ~/.bashrc,
sudo vim .bashrc
添加以下内容到最后,如:
export PATH=/usr/local/cuda-11.3/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64:$LD_LIBRARY_PATH
vim常用命令
:wq #保存并退出
:q #退出
i #在光标所在位置前插入字符
然后source一下使其生效
source .bashrc
nvcc -V
出现以下信息,则配置成功:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Sun_Mar_21_19:15:46_PDT_2021
Cuda compilation tools, release 11.3, V11.3.58
Build cuda_11.3.r11.3/compiler.29745058_0
依次执行以下命令(也可以不用删除安装包)
cd 下载
ls #看是否有cuda
rm -r cuda
换终端,装cudnn
下载上图的cudnn包 cudnn-11.3-linux-x64-v8.2.1.32.tgz
在下载目录下打开终端,依次输入
tar xzvf cudnn-11.3-linux-x64-v8.2.1.32.tgz
sudo mv cuda/lib64/* /usr/local/cuda-11.3/lib64/
sudo mv cuda/include/* /usr/local/cuda-11.3/include/
输入
nvcc -V
如下所示则表示安装成功