美文网首页
win10 64bit+Anaconda3+CUDA10+CUD

win10 64bit+Anaconda3+CUDA10+CUD

作者: 8a8d7f2e842b | 来源:发表于2019-12-04 12:54 被阅读0次

1. 下载安装Anaconda3:

  • 官网下载最新版本。安装过程省略,安装到指定位置即可。
  • 在系统环境变量path中加入anaconda的安装地址:

D:\Anaconda3
D:\Anaconda3\Scripts
Anaconda详细安装及使用教程

2. 下载安装CUDA10.0.130及CUDnn7.6.5.32

2.1 点击cuda_10.0.130_411.31_win10.exe安装,注意保证图中Current Version <= New Version。

一直点击下一步完成安装即可。

2.2 将cudnn-10.0-windows10-x64-v7.6.5.32压缩包中的文件拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0对应路径下
2.3 将对应路径加入系统环境变量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\lib64
C:\Program Files\NVIDIA Corporation\NVSMI
C:\Program Files\NVIDIA Corporation\NVIDIA NvDLISR

2.4 执行nvcc -V 和 nvidia-smi 即可看到对应的版本及进程信息

3. 创建虚拟环境,下载tensorflow-gpu

3.1 创建tensorflow虚拟环境

conda create -n tensorflow python=3.6 numpy pip

3.2 进入tensorflow环境

activate tensorflow

3.3 下载tensorflow-gpu

pip install tensorflow-gpu

3.4 安装jupyter

conda install jupyter

3.5 运行jupyter

jupyter notebook

4.案例

import tensorflow as tf
import os
# os.environ["CUDA_VISIBLE_DEVICES"] = "-1"  # 这一行注释掉就是使用gpu,不注释就是使用cpu
mnist = tf.keras.datasets.mnist
(x_train, y_train),(x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0
model = tf.keras.models.Sequential([
  tf.keras.layers.Flatten(input_shape=(28, 28)),
  tf.keras.layers.Dense(512, activation=tf.nn.relu),
  tf.keras.layers.Dropout(0.2),
  tf.keras.layers.Dense(10, activation=tf.nn.softmax)
])
model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])
model.fit(x_train, y_train, epochs=5)
model.evaluate(x_test, y_test)

执行nvidia-smi发现已成功使用GPU运行程序


相关文章

网友评论

      本文标题:win10 64bit+Anaconda3+CUDA10+CUD

      本文链接:https://www.haomeiwen.com/subject/zdgkgctx.html