anaconda下安装tensorflow2.0.0
cuda10.0下载安装
CUDA Toolkit 10.0 Download:https://developer.nvidia.com/cuda-downloads
网页最下方找到
点击下载
下载完成后,打开软件进行安装
这里最好默认安装路径
等待检查系统兼容性完成
这里需要选择自定义安装
这里只要选第一个
取消第一个里面的Visual Studio intergration 的选择
这里选择默认的路径安装(或自己一定要记住自己安装的其他路径!!!!)
安装完后就完成了。安装完成后系统变量会自动添加,可自行查看系统变量
可以在cmd中输入命令nvcc -V 查看是否安装成功
以上就是cuda成功安装全过程
安装cudnn
WIN10安装cuDNN
打开cuDNN网页:https://developer.nvidia.com/rdp/cudnn-download
没有账号的需注册账号才能下载
选择与CUDA匹配的cuDNN版本进行下载
下图中 I Agree处需要勾选上
这里一定要看好选好对应版本!
下载之后是个压缩包文件,解压缩:
复制下面所有内容:
将CUDA\bin、CUDA\include、CUDA\lib中的内容拷贝到相应的C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0文件路径下即可(因为此处我是默认路径的,此处只用把三个文件夹全部选中复制粘贴进以上路径的文件夹即可,不需一一复制,会自动找到位置):
放心点击勾选,继续 ,文件不重复!!!
接下来添加环境变量(path):C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
将添加内容上移到最顶层
安装tensorflow2.0.0
打开Anaconda Prompt
输入命令:pip install tensorflow-gpu==2.0.0 -i https://pypi.douban.com/simple
等待安装完成
接下来测试是否安装成功:
命令测试:
进入python环境:输入python
python
import tensorflow as tf
证明安装tensorflow成功
接下来测试是否能够gpu运行:
version =tf.__version__
gpu_ok = tf.test.is_gpu_available()
print("tf version",version,"\nusr GPU ",gpu_ok)
这里显示True证明安装成功!