安装cuda和cudnn
安装cuda官方文档:docs.nvidia.com/cuda/cuda-i…
安装cudnn官方文档:docs.nvidia.com/deeplearnin…
查看是否有显卡驱动
nvidia-smi
如果有输出即为安装了N卡驱动,并可以看到相应推荐的cuda版本
如果没有安装N卡驱动,可以打开软件更新管理器在驱动选项里进行安装,安装推荐的就好了
或者敲下命令自动安装
sudo ubuntu-drivers autoinstall
检验是否安装完成可以重新执行nvidia-smi
下载相应的cuda和安装
前往官网https://developer.nvidia.com/cuda-toolkit-archive下载对应版本的cuda,需要登陆。
切记一定要是与nvidia-smi显示出来与之对应的,下载run包的话可以比较方便快捷的安装完成,下载deb包可能安装过程还需要另外下载其他东西,不排除会有被墙的可能
run包下载完毕
sudo sh cuda_10.2.89_440.33.01_linux.run
-
点击 continue
-
输入 accept
-
cuda 安装包是自带显卡驱动的,所以这一步按空格去掉安装显卡驱动的选项,然后选择install
我们已经自行安装过显卡驱动了,另外这个显卡驱动安装也十分有问题,不建议在这里安装
-
安装完之后,配置环境变量
export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}首先前往/usr/local/目录查看是否有cuda-版本号这样类似的目录,切记其中的路径为你相对应安装的cuda版本的目录
-
检验
cd /usr/local/cuda/samples/1_Utilities/deviceQuery sudo make -j4 ./deviceQuery # 如果输出如下,即为安装成功 deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 11.4, CUDA Runtime Version = 11.4, NumDevs = 1 Result = PASS
安装cuDNN
前往官网https://developer.nvidia.com/cudnn下载相应版本的cudnn,需要登陆。
下载与cuda相对应版本的即可,下载tar格式的压缩包,也可以下载三个deb包
这三个deb包安装是有一定顺序的,安装出错了百度一下就知道安装哪一个先了
tar格式压缩包:
下载解压之后,将cuda/include/cudnn.h文件复制到usr/local/cuda/include文件夹,将cuda/lib64/下所有文件复制到/usr/local/cuda/lib64文件夹中,并添加读取权限:
sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
检验
# Copy the cuDNN samples to a writable path.
cp -r /usr/src/cudnn_samples_v8/ $HOME
# Go to the writable path.
cd $HOME/cudnn_samples_v8/mnistCUDNN
# Compile the mnistCUDNN sample.
make clean && make
# Run the mnistCUDNN sample.
./mnistCUDNN
# If cuDNN is properly installed and running on your Linux system, you will see a message # similar to the following:
# Test passed!
自此,所有安装都已经完成。
nux system, you will see a message # similar to the following:
Test passed!
自此,所有安装都已经完成。