我創建了一個 virtualenv 并安裝了 TensorFlow 2.4.0。CUDA 已經安裝,所以我嘗試按照這些說明安裝 Cudnn
$ tar -xzvf cudnn-11.0-linux-x64-v8.0.2.39.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*
確實在安裝檔案安裝到這些路徑 /usr/local/cuda/include 和 /usr/local/cuda/lib64 之后。這就是結果(串列斷裂)

雖然檔案“libcudnn.so.8”存在,但它無法識別

由于創建了虛擬環境,我應該將檔案復制到另一個檔案夾嗎?淺綠色檔案表示什么?
uj5u.com熱心網友回復:
我只是在 .bashrc 中添加了這個
export PATH=/usr/local/cuda-11.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH
根據這個 在 Linux (Ubuntu 20.04) 上安裝 NVIDIA 驅動程式、CUDA 和 cuDNN
轉載請註明出處,本文鏈接:https://www.uj5u.com/gongcheng/524644.html
標籤:张量流显卡库丁
