安装cuda

https://developer.nvidia.com/cuda-11.1.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=2004&target_type=runfilelocal

安装cudnn

https://developer.nvidia.com/rdp/cudnn-archive#a-collapse804-111
https://zhuanlan.zhihu.com/p/339062791

  1. tar -xzvf cudnn-11.1-linux-x64-v8.0.4.30.tgz
  3. sudo cp cuda/include/* /usr/local/cuda/include/
  4. sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
  5. sudo chmod a+r /usr/local/cuda/include/cudnn.h
  6. sudo chmod a+r /usr/local/cuda/lib64/libcudnn*