Win10系统下 CUDA9.0+cuDNN7.4+Python3.6+Pytorch-gpu+tensorflow-gpu安装

这个问题我遇到不少坑。网上类似的教程很多,我总结一个很实用的方法(避免入坑)

硬件配置:我的电脑配置是win10系统(64bit),处理器i7-8750,显卡8G GTX1070

1. 首先安装CUDA9.0和cuDNN7.4

官网下载CUDA9.0和cuDNN7.4(对应版本),确保你的电脑有GPU(需要查一下GPU可安装的CUDA和CUDnn版本)

cuda官网:https://developer.nvidia.com/cuda-toolkit-archive

cudnn官网:https://developer.nvidia.com/rdp/cudnn-download(需注册)

安装CUDA9.0和cuDNN7.5之前需要安装vs2015(或者更高版本),不然会出错(vs2015我最后是用了一个社区版的,可以去网上下载,我自己找了好几个版本的,有的版本在win10下安装不成功win7下安装却可以)

双击从官网下载的cuda9.0安装程序,解压得到一个文件夹,然后自动开始安装程序,按照提示安装即可(最好使用管理员权限进行安装)。安装cuda9.0如果出现些问题,可以尝试重新安装下NVIDIA显卡驱动(一般问题不大,关键是要安装vs2015)。

安装cuda9.0后,要在用户变量的Path中添加:C:\ProgramData\NVIDIA GPU Computing Toolkit\v9.0\bin  ,同时确保系统变量中 CUDA_PATH和CUDA_PATH_V9.0已经存在。完成操作后打开命令提示符,输入 nvcc  --version,查看cuda安装是否成功

接着安装cudnn,解压cudnn7.4安装包,将文件夹里内容拷贝至安装CUDA的地方覆盖对应文件夹,默认文件夹在: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

这里再分享一个windows系统下cuda版本切换的方法:比如已经安装了cuda8.0和cuda9.0(两个版本都是一样的安装步骤),到系统环境变量Path编辑,可以看到

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\libnvvp

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

比如当前是cuda8.0,将v9.0的两个路径上移到顶部,就可以切换到cuda9.0。可以重新开启命令行输入 nvcc -V c 查看,这大概是因为环境变量先找到 v9.0 吧,同样想切换到cuda8.0,把v8.0 的两个路径上移到顶部即可。(前提cuda8.0和9.0都安装好了)

在Windows系统中安装了NVIDIA的驱动后,会发现不能使用nvidia-smi命令,把nvidia-smi.exe的路径添加到系统环境变量(默认路径为C:\Program Files\NVIDIA Corporation\NVSMI),然后重新打开cmd即可

2. 安装python3.6和pytorch

CUDA9.0和cuDNN7.4安装好之后,接着安装python和pytorch。

到python官网下载 python-3.6.5-amd64,网址:https://www.python.org/downloads/windows/

说明一下,很多网上安装教程推荐anaconda3,anaconda有强大的软件包管理功能并自带多个常用python库,可以说是入门python的利器(确实很好用),但我个人在使用过程遇到一些奇怪的问题(折腾起来也很麻烦)。这里就直接安装python3.6,(记得安装选项中勾上pip和添加环境变量),然后利用pip安装numpy,pandas,matplotlib,scipy等python库。

最后要安装pytorch了。因为要使用GPU (深度学习利器),我们要下载gpu版本的pytorch。去pytorch官网下载

pytorch网站:https://pytorch.org/  选择stable,windows,pip,python3.6,cuda9.0选项下的whl文件下载。注意最好不要修改whl文件名(会可能导致pytorch无法安装,我被这么坑过)。pytorch1.0下载网址:https://download.pytorch.org/whl/cu90/torch-1.0.1-cp36-cp36m-win_amd64.whl

3. 最后torch安装成功,接着 pip install torchvision (torchvision 是个很重要的模块,最好装上),这样pytorch就安装完成。不过安装成功后导入torch模块( import torch )也可能会出现错误(很少见),这时怎么办

去 https://anaconda.org/anaconda/intel-openmp/files 下载 win-64/intel-openmp-2019.1-144.tar.bz2(最近的版本,win-64位),解压 intel-openmp,将Library\bin下的dll文件全都拷贝到Python安装目录里的dll路径里(\Python36\DLLs)

然后把DLLs文件夹路径添加到环境变量中(我的电脑--属性--高级系统设置--高级--环境变量--path--添加)

完美解决!!!

成功后测试一下cuda是否可以用,python脚本中输入:  print(torch.cuda.is_available())

顺便看看pytorch版本,python脚本中输入:  print(torch.__version__)

4. 安装tensorflow-gpu,这个比较简单,可以先下载好相应的whl文件在本地安装,下载地址如下:

https://pypi.org/project/tensorflow-gpu/#files

我这里下载了支持python3.6的版本 tensorflow_gpu-1.12.0-cp36-cp36m-win_amd64.whl,命令行使用pip进行安装

pip install tensorflow_gpu-1.12.0-cp36-cp36m-win_amd64.whl

最后检查一下tensorflow是否可用(一般不会有问题),启动一个python脚本,输入

import tensorflow as tf

print(tf.__version__)

没有问题即安装成功

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章