第一步:查Linux中,与cuda9.0 对应的tensorflow版本
这里查更多版本信息
我的是cuda9.0 , GCC4.8,就只能装tensorflow_gpu-1.12.0咯
第二步:使用anaconda命令创建python的虚拟环境
conda create -n env_name python=3.6
然后激活环境
conda activate env_name
第三步:加豆瓣源下载
pip install tensorflow-gpu==1.12.0 -i https://pypi.douban.com/simple/
cmd里查看是否安装成功
python
import tensorflow
print(tensorflow.__version__)
yes,成功啦啦!