win10安装tensorflow-gpu1.13.1+cuda10.0+cudnn7.3.1
程序员文章站
2023-11-19 16:09:52
一,本机配置 Win10 64bit NVIDIA GeForce GTX 960M Python3.7(Anaconda) 二,安装CUDA 亲测,TensorFlow-gpu1.13.1支持cuda10.0的版本,所以我们可直接选择cuda10.0的版本 Window10下载CUDA10 安装步 ......
一,本机配置
win10 64bit
nvidia geforce gtx 960m
python3.7(anaconda)
二,安装cuda
亲测,tensorflow-gpu1.13.1支持cuda10.0的版本,所以我们可直接选择cuda10.0的版本
安装步骤为:
1)运行.exe可执行和层序。安装软件会先运行一个系统检查,如果没有软硬件不兼容的情况就能继续进行下一步。若有不兼容情况,系统检查则会报错,无法进行下一步。如果报错了,就最先考虑下是不是自己的显卡不被该版本支持。在安装过程中,它会先自动安装和你电脑对应的nvidia驱动。
2)选择默认安装就行。
三,安装cudnn
下载cudnn需要自己先在以上网址(cudnn官网)注册个账号,注册成功后登陆进去选择cudnn7.3.1下载即可。
下载完成后是一个.zip包,解压后进入目录,会发现三个子目录/bin,/include,/lib/x64。进去后会发现有三个文件,将这三个文件分别拷贝到cuda的安装目录的对应子目录下
默认cuda的安装目录是c:\program files\nvidia gpu computing toolkit\cuda
四,配置系统path环境变量
检查你的系统path环境变量中是否有这四项,没有请手动添加
c:\program files\nvidia gpu computing toolkit\cuda\v10.0\bin c:\program files\nvidia gpu computing toolkit\cuda\v10.0\libnvvp c:\program files\nvidia gpu computing toolkit\cuda\v10.0\lib\x64 c:\program files\nvidia gpu computing toolkit\cuda\v10.0\extras\cupti\libx64
五,安装tensorflow-gpu
打开anaconda prompt,输入
pip install tensorflow-gpu
会自动下载并安装tensorflow-gpu
六,安装测试
打开pycharm,创建一个test.py文件,输入
import tensorflow as tf tf.enable_eager_execution() print(tf.__version__) hello = tf.constant("hello tensorflow") print(hello.numpy())
出现下列结果表明安装成功
上一篇: 关于debug
下一篇: JavaScript 数组方法