Step1
去到NVIDIA 官网下载 CUDA Toolkit 10.0
CUDA Toolkit 10.0 Archive
选择自定义
自定义
如果没有安装Visual Studio 可以把该项去掉✔
image.png
注意 如果新版本 < 当前版，注意要取消✔
image.png
Step2
下载 Download cuDNN v7.6.2 (July 22, 2019), for CUDA 10.0
cuDNN v7.6.2 (July 22, 2019), for CUDA 10.1
下载解压之后放在 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0 路径下并且改名为 cudnn
image.png
Step3
配置系统环境变量
系统环境变量
extras\CUPTI\libx64
cudnn\bin
Step4
检测nvcc
nvcc -V
nvcc
Step5
测试 Tensorflow
安装TensorFlow2.0
pip install tensorflow==2.0.0-rc0
在终端启动 ipython
ipython
ipython
导入tensorflow
import tensorflow as tf
import tensorflow
测试是否支持GPU
tf.test.is_built_with_gpu_support
tf.test.is_built_with_gpu_support
测试运算
tf.constant(1.) + tf.constant(2.)
测试运算
网友评论