1.首先安装Anaconda(用来管理下载的环境)
官网下载链接 :https://www.anaconda.com/distribution/#download-section
选择win 64位下载即可。
下载后exe安装,可能需要注意的是会询问你是否添加到环境变量中,可以选择不添加,以后开Anaconda可以用开始里面的Anaconda Prompt打开。
2.安装CUDA10.0&cuDNN
教程https://blog.csdn.net/qq_37296487/article/details/83028394
CUDA10.0官方下载地址https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
注意选择local本地安装
参考https://blog.csdn.net/weixin_51130521/article/details/119874301
3.安装tensoflow-gpu
打开Anaconda prompt,创建tensorflow环境,命令:
conda create -n tensorflow python=3.6
- 使用conda info -envs查看一下刚刚创建的tensorflow环境
应该可以看到一个base,一个tensorflow(在anaconda3\envs中)
2.使用
activate tensorflow
激活tensorflow环境,这样就从base中切换到tensorflow中了,命令行的提示会相应更改。
3 .此时在tensorflow的环境下安装tensorflow-gpu 1.14.0,因为网络可能较慢,这里选择换源下载,代码如下:
pip install --ignore-installed --upgrade -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.14.0
4.pycharm使用tensorflow环境
- 在settings->project interpreter点击小齿轮,add一个环境,点下面的existing environment(不要new,不然会有问题),选择路径就是前面tensorflow目录下的python.exe