anaconda安装gpu版本的tensorflow2.0.0教程

anaconda安装gpu版本的tensorflow教程

1、安装anaconda
2、创建虚拟环境
3、安装tensorflow及相关包
4、测试是否运行正常

1、安装anaconda
直接百度,教程很简单

2、创建虚拟环境
打开anaconda的anaconda prompt,输入以下conda命令:
conda create -n tensorflow2.0gpu python=3.6
跳出询问是否下载后,输入y下载

3、安装tensorflow及相关包
(1)查看自己的显卡是否支持gpu加速,一般算力要大于3.5
(2)打开命令行cmd,输入nvidia-smi,NVIDIA驱动程序需大于410.x版本
anaconda安装gpu版本的tensorflow2.0.0教程
(3)打开anaconda prompt,输入activate tensorflow2.0gpu**创建的虚拟环境,分别输入conda install cudatoolkit;和conda install cudnn=7.6.0下载好这两个包
anaconda安装gpu版本的tensorflow2.0.0教程
(4)直接下载gpu版本的tensorflow
pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

4、测试是否运行正常
在自己的编辑器中,配置好环境,
import tensorflow as tf
tf.version #查看版本
tf.test.is_gpu_available() #查看gpu是否可用

通过以上测试基本ok,但是后续运行中可能会遇到cudnn无法加速的问题,一般原因是显卡已经爆了,具体再百度。
有问题欢迎咨询qq1269143766,同时执行各种深度学习代做。