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