深度學習 環境相關(持續更新)

3. cudnn升級

最近tensorflow-gpu版本升級了, 原來要求使用cudnn5, 現在需要cudnn6, 于是乎到nvidia官網重新安裝, 下載cudnn-8.0-linux-x64-v6.0.tgz這個包, 解壓 tar -xvf cudnn-8.0-linux-x64-v6.0.tgz, 分別拷貝文件到對應目錄

sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include
sudo cp cuda/lib64/libcudnn.so.6.0.21 /usr/local/cuda-8.0/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn*
cd /usr/local/cuda-8.0/lib64
sudo ln -sf libcudnn.so.6.0.21 libcudnn.so.6
sudo ln -sf libcudnn.so.6 libcudnn.so  
sudo ldconfig  

4. tensorflow

tensorflow輔助
$ pip install tensorboard

注意在virtualEnv下安裝有的bug, 是個virtualEnv的'site' has no attribute "getsitepackages"的bug, 暫時用最粗暴的方式解決, 在外面安裝一個tensorflow和tensorboard

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容