一、cudnn7.0.5安裝包
CuDNN是NVIDIA Deep Learning SDK的一個組成部分,提供了高性能的GPU加速深度神經網絡庫。cudnn7.0.5是較久以前的版本,但在某些場景下,該版本仍然是不可替代的。在安裝cudnn7.0.5之前,我們需要下載相應的安裝包。
// 從官網上下載
https://developer.nvidia.com/rdp/cudnn-archive
當然,在國內,我們也可以通過百度網盤或其他途徑下載到安裝包。
二、cudnn7.0.5安裝(Windows)
cudnn7.0.5安裝包下載完成後,我們可以開始進行安裝。下面是Windows安裝的步驟。
1. 解壓安裝包
tar -zxvf cudnn-7.0.5-win-x64-v5.1.zip
2. 複製文件
打開目錄,將相應的文件複製到你的CUDA目錄。請注意,你需要將相應的文件放到正確的位置。具體的位置可能因你的CUDA安裝路徑而異。
cp cuda/include/* /usr/local/cuda/include/
cp cuda/lib64/* /usr/local/cuda/lib64/
3. 驗證
重啟你的PyTorch或TensorFlow,然後執行以下命令驗證cudnn是否正確安裝。
import torch
torch.backends.cudnn.version()
如果不出意外,你應該能夠看到類似以下的輸出,說明cudnn已經被安裝並成功運行。
7005
三、cudnn7.0.5安裝(Linux)
在Linux上安裝cudnn7.0.5也很簡單。請按照以下步驟操作。
1. 解壓安裝包
tar -zxvf cudnn-7.0.5-linux-x64-v5.1.tgz
2. 複製文件
打開目錄,將相應的文件複製到你的CUDA目錄。請注意,你需要將相應的文件放到正確的位置。具體的位置可能因你的CUDA安裝路徑而異。
sudo cp cuda/include/* /usr/local/cuda/include/
sudo cp cuda/lib64/* /usr/local/cuda/lib64/
3. 更新庫並添加環境變量
更新系統庫並添加環境變量,以確保cudnn能夠正確運行。
sudo ldconfig /usr/local/cuda/lib64
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
4. 驗證
重啟你的PyTorch或TensorFlow,然後執行以下命令驗證cudnn是否正確安裝。
import torch
torch.backends.cudnn.version()
如果不出意外,你應該能夠看到類似以下的輸出,說明cudnn已經被安裝並成功運行。
7005
四、cudnn7.0.5的下載
cudnn7.0.5可以直接從NVIDIA官網上下載。在下載前,請確保你的CUDA版本與cudnn7.0.5兼容。可以通過以下命令查看你的CUDA版本:
nvcc --version
五、cudnn7.0.5的百度網盤下載
如果你無法從官網上下載cudnn7.0.5,你也可以使用百度網盤下載。以下是cudnn7.0.5的百度網盤地址。
鏈接: https://pan.baidu.com/s/1Jfs8eLZlM-gQdvZ3ZvWdlQ
提取碼: 5l7t
需要注意的是,由於網盤的不穩定性,該鏈接隨時可能失效。
原創文章,作者:UGRK,如若轉載,請註明出處:https://www.506064.com/zh-hant/n/146338.html