win10_64位+GTX1070max-q安装CUDA10.0.130+CUDNN7.4.2+Anaconda3.5.2+Tensorflow-gpu2.0.0+PyTorch-gpu1.2.0

安装特定版本NVIDIA显卡驱动

从CUDA-wiki百科中可以查到各个版本的CUDA支持的GPU计算等级范围。网址:https://en.wikipedia.org/wiki/CUDA#GPUs_supported 。然后打开网址:https://developer.nvidia.com/cuda-gpus ,查看显卡对应的计算等级(Compute Capability)。
GTX1070max-q的核心与GTX1070的核心相同,计算等级都为6.1。由此可知我们的GTX-1070MAX-q显卡可以安装CUDA10。关于CUDA各版本对nvidia驱动版本的要求可以看这里:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html 。
这里我们选择安装CUDA10.0.130。该版本CUDA在Linux x86_64系统上需要>= 410.48版本的NVIDIA驱动,在windows上需要>= 411.31版本的NVIDIA驱动。由于我的电脑win10系统已经升级到了441.87NVIDIA驱动,因此符合要求。如果你的驱动不符合版本要求,可以在这里下载特定版本的NVIDIA驱动:https://www.nvidia.cn/Download/Find.aspx?lang=cn 。

下载和安装CUDA10.0.130

我们在这个网址可以查到各个版本的tensorflow支持的CUDA和CUDNN版本:https://www.tensorflow.org/install/source#common_installation_problems 。我们要安装tensorflow2.0.0,CUDA10.0.130支持该版本的tensorflow,其下载地址:https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10 。我们可以选择直接下载完整安装包。
下载后按默认设置安装即可。
安装完成后,添加下面四个路径到用户环境变量中:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp
这四个路径非常重要,如果不添加,会出现import tensorflow时正常不会报错,但当你要执行tf.Session()的时候就会卡住。因为这个时候将会调用cuda,没有路径的话cuda需要的各种lib、dll加载不了。
上述操作全部完成后打开cmd,输入nvcc -V,如果出现CUDA版本信息,证明路径已经配置好了。

下载和安装CUDNN7.4.2

由这个网址:https://www.tensorflow.org/install/source#common_installation_problems 。我们还可以查到各个版本的tensorflow需要安装的CUDNN版本。我们要安装tensorflow2.0.0,CUDNN7.4.2支持该版本,其下载地址:https://developer.nvidia.com/rdp/cudnn-archive 。
下载文件。解压缩后得到一个文件夹。将这个目录中所有文件复制粘贴到CUDA安装位置:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0即可。

下载和安装Anaconda3.5.2

我们可以从anaconda官网上下载anaconda,如果觉得速度慢的话也可以从这个镜像:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 中下载也可以,注意安装anaconda3-5.2.0,这个版本的anaconda默认安装python3.6。
安装过程按默认设置下一步即可。注意不要勾选add anaconda to the system PATH environment variable,我们待会儿会手动设置。
安装完成后,在用户环境变量中新建一个path,添加下面三个路径:
C:\Users\zgcr6\Anaconda3
C:\Users\zgcr6\Anaconda3\Scripts
C:\Users\zgcr6\Anaconda3\Library\bin
然后打开cmd,输入conda -V测试一下,如果能正常显示版本号,说明已经配置好了。

安装Tensorflow-gpu2.0.0和pytorch1.2.0

我们在anaconda中先新建一个名为tensorflow-gpu-2.0.0-pytorch1.2的环境,该环境安装python3.6。打开anaconda prompt,使用命令:

//首三行代码是将conda安装源换成清华源(永久设置),因为官方源是pypi,在国内下载速度太慢了
//清华源是官网pypi的镜像,每隔5分钟同步一次,地址为 https://pypi.tuna.tsinghua.edu.cn/simple
//如果使用pip时想临时用一下清华源,使用命令:pip install -i https://pypi.tuna.tsinghua.edu.cn/simple package-name
//还可以使用阿里源:https://mirrors.aliyun.com/pypi/simple
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
# 如果需要换回conda默认源
# conda config --remove-key channels
conda search tensorflow-gpu //查找tensorflow-gpu的所有版本
conda create -n tensorflow-gpu-2.0.0-pytorch1.2 python=3.6.6 //新建一个名为tensorflow-gpu-2.0.0-pytorch1.2的环境

然后等待安装完成。该命令创建了一个名为tensorflow-gpu-2.0.0-pytorch1.2的环境,并安装了python3.6.6。创建的环境位置在:C:\Users\zgcr6\Anaconda3\envs。
然后我们将下面这两个目录添加到用户变量中,注意要放到anaconda的目录之前:
C:\Users\zgcr6\Anaconda3\envs\tensorflow-gpu-2.0.0-pytorch1.2
C:\Users\zgcr6\Anaconda3\envs\tensorflow-gpu-2.0.0-pytorch1.2\Scripts
然后打开cmd,使用下面的命令安装tensorflow2.0.0:

python -m pip install tensorflow-gpu==2.0.0
//如果觉得慢可以临时用一下清华源
python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.13.1

等待安装完成即可。
安装完成后,我们还需要验证tensorflow能否正常调用显卡进行计算。打开cmd.exe,输入python,然后依次输入下面的命令:

import tensorflow as tf
hello = tf.constant("hello, world!")
print(hello)

当输入到第三句sess = tf.Session()时,即开始调用显卡。注意如果是第一次在系统中使用tensorflow来调用显卡时,在adding visible gpu devices:0这里会卡住3-5分钟,之后如果能够继续运行并正常输出hello world,则说明安装成功。
再安装pytorch1.2.0。

pip install torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

安装完成后,我们要验证一下安装是否成功。在cmd窗口中输入python,然后依次输入下列命令:

import torch
print(torch.cuda.is_available())

如果输出为True,则说明我们的安装成功了。

如何使用我们配置的tensorflow-gpu-2.0.0-pytorch1.2的环境

在cmd中使用tensorflow-gpu-2.0.0-pytorch1.2环境

将tensorflow-gpu-2.0.0-pytorch1.2环境的路径加入到用户环境变量中,即路径C:\Users\zgcr6\Anaconda3\envs\tensorflow-gpu-2.0.0-pytorch1.2,注意这个路径一定要放在我们刚才添加的路径C:\Users\zgcr6\Anaconda3之前。
此时打开cmd后,输入python运行的就是C:\ProgramData\Anaconda3\envs\tensorflow-gpu-2.0.0-pytorch1.2目录中的python.exe。

在编译器(如pycharm)中使用tensorflow-gpu-2.0.0-pytorch1.2环境

我们以pycharm为例,在pycharm中找到项目的解释器设置(file->settings->project interpreter),将解释器路径设为我们的C:\Users\zgcr6\Anaconda3\envs\tensorflow-gpu-2.0.0-pytorch1.2\python.exe。

  • 2
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值