电脑配置:i7-8700K 16G 256GSSD+2T GTX1080Ti
由于要用tf==0.12.1,所以配置的深度学习环境并不是最新的。想安装最新的TF可以参考此教程
1、win10+Ubuntu16.04双系统,安装教程、补充教程。
2、cuda8.0+cudnn5.1+anaconda 安装教程,建议从作者提供的百度云链接下载,英伟达官网链接有点慢。
注:安装英伟达的驱动,桌面>>系统设置>>软件和更新>>附加驱动>>选择英伟达的驱动。
注:cuda的例子没有下载,所以教程中没有make例子。
3、用anaconda安装TF,参照tensorflow官网
conda create -n tensorflow ##后面加 pip python=2.7 (or 3.X)
#Activate the conda environment by issuing the following command:
source activate tensorflow
(tensorflow)$ # Your prompt should change
(tensorflow)$ pip install --ignore-installed --upgrade \ tfBinaryURL ##轮子地址
也可以直接 pip install tensorflow-gpu==0.12.1
激活conda环境后,进入python
检验TF和cuda是否安装成功
# Python
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
我的一开始报错,
I tensorflow/stream_executor/dso_loader.cc:119]Could’t open CUDA library libcudnn.so.LD_LIBRARY_PATH:
I tensorflow/stream_executor/cuda/cuda_dnn.cc3459]Unable to load cuDNN DSO
卸载tf安装cpu版本,可以跑通代码,紧接着在安装,tf-gpu,结果又成功了!!!
结果可能是cudnn没用上,感觉跑的挺慢的。
想安装最新的TF可以参考此教程