conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
conda config --show channels
conda create -n python=3.7
conda activate tensor_susan
*如果需要退出该环境,回到base,则输入命令
conda deactivate
可以在github上直接下载支持的tensorflow gpu的.whl文件,再用pip安装
github链接
https://github.com/fo40225/tensorflow-windows-wheel
对应找到py37, CUDA v10, cuDNN 7.3版本的tensorflow,下载windows版本
点进去后找到
下载到本地
手动安装(pip)tensorflow
打开Anaconda Shell Prompt,进入刚才创建的tensorflow虚拟环境。
conda activate
cd到刚才下载whl文件的目录。
cd <安装了whl文件的路径>
用pip安装:
pip install tensorflow_gpu-1.12.0-cp37-cp37m-win_amd64.whl
直接输入命令即可
conda install tensorflow-gpu
(安装的是py37, tensorflow gpu 1.14.0版本)
\envs\\python.exe
然后点OK,环境配置完成。
import tensorflow as tf
with tf.device('/cpu:0'):
a = tf.constant([1.0, 2.0, 3.0], shape=[3], name='a')
b = tf.constant([1.0, 2.0, 3.0], shape=[3], name='b')
with tf.device('/gpu:1'):
c = a + b
# 注意:allow_soft_placement=True表明:计算设备可自行选择,如果没有这个参数,会报错。
# 因为不是所有的操作都可以被放在GPU上,如果强行将无法放在GPU上的操作指定到GPU上,将会报错。
sess = tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True))
# sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
sess.run(tf.global_variables_initializer())
print(sess.run(c))
运行后,如果能看到gpu信息,则表示配置成功
至此,Tensorflow安装并检验完成。
Keras安装官方说明
克隆tensorflow虚拟环境,并命名好新的Keras环境并激活
conda create -n keras_susan --clone tensor_susan
conda activate keras_susan
pip install keras
或
conda install keras-gpu
如果输入conda install keras会装成依赖cpu的
Ps: 之前我也不知道为什么创建了一个python 3.6版本的keras虚拟环境,然后import tensorflow或者import keras都会报错:
No module named '_pywrap_tensorflow_internal'
上网查阅资料过后,发现应该是CUDA,cuDNN版本与Python版本的兼容问题。最后重新用python3.7搭建虚拟环境后,直接conda install tensorflow-gpu,然后conda install keras-gpu就安装好了,没有再出现问题。
RTX2060 —— Anaconda + CUDA 10.0 +Cudnn 7.3 + Tensorflow-GPU + Keras 安装(一)Anaconda3
RTX2060 —— Anaconda + CUDA 10.0 +Cudnn 7.3 + Tensorflow-GPU + Keras 安装(二)CUDA10.0 & cuDNN 7.3
RTX2060 —— Anaconda + CUDA 10.0 +Cudnn 7.3 + Tensorflow-GPU + Keras 安装(三)Tensorflow + Keras
RTX2060 —— Anaconda+CUDA+Cudnn+Pytorch安装(四)Pytorch安装