tensorflow-gpu == 2.0.0-beta1安装说明

1、安装步骤说明

• 创建环境

conda create -n TF2.0 python=3.7

• 激活环境

conda activate TF2.0

• 安装cudatoolkit

conda install cudatoolkit=10.1

• 安装cudnn

conda install cudnn=7.6

• 安装tensorflow-gpu == 2.0.0-beta1

pip install tensorflow-gpu == 2.0.0-beta1 -i https://pypi.douban.com/simple

2、测试gpu

import tensorflow as tf
print('GPU',tf.test.is_gpu_available())
a = tf.constant(2.)
b = tf.constant(4.)
print(a*b)

3、注意事项

• 建议都加上豆瓣源下载，这样下载速度飞快
• 但是，豆瓣源的包没有清华源的包多，不过，豆瓣源很多时候也很够用啦
• 建议安装numpy版本不要太高，可以使用numpy==1.16.5

pip install numpy==1.16.5 -i https://pypi.douban.com/simple