cuda 配置要点

时间:2023-03-09 04:30:36
cuda 配置要点

1. 安装驱动 :sudo apt-get install nvidia-

2. 安装cuda : cuda 文件中包含驱动程序,因此在安装过程中当被问及是否安装驱动时,选择no

3. 安装cudnn: 下载cudnn, 执行一下命令

tar -xzvf cudnn-8.0-linux-x64-v6.0.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 /usr/local/cuda/lib64/libcudnn*

注 参考作者:Louiseluke