先看本文最底部的方法!
这一个可以看腾讯云的文档,安装cuda
https://cloud.tencent.com/document/product/560/8064
然后是安装cudnn
https://developer.nvidia.com/cudnn
可以参考这个:
https://blog.csdn.net/plc19981102/article/details/120807321
tar -xzvf cudnn-10.2-linux-x64-v7.6.5.32.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
目前我发现最简单的方式:
apt list nvidia*
sudo apt-get install nvidia-driver-535 #选择最新的安装驱动就行
sudo apt install nvidia-cuda-toolkit