1. 程式人生 > >cuda 配置要點

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