docker容器安裝gpu環境(適用於其它linux環境)

1.下載並安裝anaconda環境(官網太慢)

https://www.jianshu.com/p/e298b9d3afae

2.安裝cuda、cudnn、tensorflow-gpu

https://www.cnblogs.com/LuckBelongsToStrugglingMan/p/12695156.html

**注意版本匹配**

在這裏插入圖片描述

3.查看安裝結果
cuda 版本 
cat /usr/local/cuda/version.txt

cudnn 版本 
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

其它參考鏈接:
https://blog.csdn.net/qq_28822933/article/details/83929960

劃重點!!

4.以上搞那麼多既複雜又容易出錯,推薦使用docker鏡像

docker pull tensorflow/tensorflow:1.8.0-devel-gpu-py3
然後在DockerFile裏面使用from tensorflow/tensorflow 進行引入使用

https://www.cnblogs.com/deepllz/p/9009486.html

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章