win10下安装cuda和cudnn

Step 1:安装CUDA 9.0 

 上图是选择win10 64位 cuda9.0本地安装文件下载的截图,下载后点击运行cuda_9.0.176_win10.exe,安装时会让你选择一个文件夹作为抽取文件临时存放,记住这不是安装文件夹,安装文件夹系统会自动安装到C盘,临时文件夹安装完后会自动删除。

 

安装完之后CUDA_PATH_V9_0和CUDA_PATH这两个系统变量会自动为你添加上。

环境变量配置: 
cuda9.0安装完成之后在系统环境变量中自动配置了两个系统变量 
CUDA_PATH:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 
CUDA_PATH_V9_0:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 
 

Step 2:安装CUDNN 7.0

下载cudnn时需要注册填写信息,选择支持cuda9.0的cudnn,我选择的是cudnn7.0.5,    7.1可能有些不兼容.下载完,解压cudnn,会出现四个文件,如下:

 在cuda安装完后,系统自动在C盘的Program Files文件中创建了NVIDIA GPUComputing Toolkit文件夹,你可以找到如下路径:

你可以发现在这个v9.0目录下也有类似cudnn中的文件夹。将cudnn的bin,include,lib中的内容 拷贝到v9.0的bin,include,lib对应目录下,提示替换的话点击OK。完成,最好重启一下电脑。OK。

此外在你环境变量PATH中添加如下内容:

 
  1. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\libnvvp

  2. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin

  3. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib

  4. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include

安装完毕后测试cuda版本:

打开cmd,输入如下指令:

nvcc -V

Step 3 Anaconda下安装Tensorflow-GPU 1.12.0

前提是你已经安装好了anaconda,点击Anaconda Prompt ( 类似cmd窗口);

新建一个python=3.6的环境(由于win10下tensorflow只支持python3)

(base) C:\Users\chens>conda create --name TFGPUP3.6 python=3.6

上面指令是创建了一个环境名为TFGPUP3.6的环境,该环境中python版本为3.6

随后会下载相应依赖库文件,安装python3.6;待环境创建完毕,激活环境;

(base) C:\Users\chens>conda activate TFGPUP3.6

激活后,命令提示行会以(TFGPUP3.6) 开头,然后在该环境中安装tensorflow-gpu ,conda无安装包,用pip 如下图:

(TFGPUP3.6) C:\Users\chens>pip install --ignore-installed --upgrade tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

命令中的链接是清华镜像,下载对应的资源速度较快。 

 如上图,是安装相应的依赖库,包文件。最后提示Successfully installed absl-py-0.7.0 astor-0.7.1 gast-0.2.2 grpcio-1.18.0 h5py-2.9.0 keras-applications-1.0.6 keras-preprocessing-1.0.5 markdown-3.0.1 numpy-1.16.0 protobuf-3.6.1 setuptools-40.7.0 six-1.12.0 tensorboard-1.12.2 tensorflow-gpu-1.12.0 termcolor-1.1.0 werkzeug-0.14.1 wheel-0.32.3

验证至看到显卡信息则成功,输入如下代码:

 
  1. >python

  2. >>>import tensorflow as tf

  3. >>>sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

 输出一些显卡的信息以及设备号:

使用下列代码测试TF安装正确性

 
  1. >>> import tensorflow as tf

  2. >>> hello = tf.constant('Hello, TensorFlow!')

  3. >>> sess = tf.Session()

  4. >>> print(sess.run(hello))

Win10下的nvidia-smi在 C:\Program Files\NVIDIA Corporation\NVSMI 目录内。

可以使用该目录下的nvidia-smi查看显卡使用状态(如下图指令);

参考:

https://blog.youkuaiyun.com/u010103202/article/details/79620551

https://blog.youkuaiyun.com/jacke121/article/details/78205845

https://blog.youkuaiyun.com/u010440456/article/details/82424894

https://blog.youkuaiyun.com/xuefengyang666/article/details/79422012

https://blog.youkuaiyun.com/gyp2448565528/article/details/79451212

https://www.cnblogs.com/sorex/p/7615185.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值