Win10下安装Anaconda+CUDA+cudnn+TensorFlow+keras+PyTorch+Pycharm

[Toc]

从零开始搭建Win10深度学习环境

image
  • 系统:Windows10
  • Anaconda:Anaconda3-5.3.1
  • CUDA版本:CUDA 9.0
  • cudnn版本:cudnn-9.0-windows10-x64-v7.4.1.5
  • Tensorflow版本:Tensorflow-gpu 1.12.0
  • keras版本:keras-2.2.4
  • Pytorch版本:Pytorch-gpu 1.0.1
  • Pycharm版本:JetBrains PyCharm Community Edition 2018.3.2 x64

Anaconda配置流程

1. 安装Anaconda

官方下载地址:[https://repo.continuum.io/archive/]
下载相应版本的Anaconda,这里使用Anaconda3-5.3.1-Windows-x86_64.exe版本。若官网下载太慢,可以使用国内的清华镜像 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

2. 添加conda镜像

TUNA 还提供了 Anaconda 仓库的镜像,运行以下命令:

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

即可添加 Anaconda Python 免费仓库。

3. 添加pip镜像

pypi 镜像使用帮助,pypi 镜像每 5 分钟同步一次。

临时使用

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

注意,simple 不能少, 是 https 而不是 http

设为默认

升级 pip 到最新的版本 (>=10.0.0) 后进行配置:

pip install pip -U
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

4. 修改Jupyter Notebok的默认工作路径

进入Anaconda Prompt,输入


image
jupyter notebook --generate-config

找到 jupyter_notebook_config.py 的路径并打此文件。


image

找到 c.NotebookApp.notebook_dir 这个变量,将你希望的路径赋值给这个变量,并删除这一行前面的“#”。


image

然而,如果你直接通过 Jupyter Notebook 的快捷方式进入,默认目录还是原来那个。如果需要修改,还需要进行如下步骤:


image

找到快捷方式,右键打开属性,将“目标”最后面的 “%USERPROFILE%” 删除就可以了。

5. 添加Anaconda环境变量

我的电脑-系统属性-高级系统设置-高级-环境变量-系统变量-找到Path

D:\Anaconda3\;
D:\Anaconda3\Scripts

6. jupyter中添加conda虚拟环境

首先安装ipykernel:

conda install ipykernel

在terminal下执行命令行

conda create -n tensorflow python=3.6 ipykernel

CUDA和cudnn配置流程

1. 查看计算机显卡型号:

在桌面电脑图标上点击右键,选择管理。


image

点击设备管理器。


image

点击显示适配器。
image

如图即可看到当前安装的显卡型号。


显卡型号.png

例如本机显卡型号为GTX1060

2. 查看显卡是否支持CUDA版本

去CUDA官网查看计算机显卡支持的版本型号
https://developer.nvidia.com/cuda-gpus

gtx1060.png

我们可以看到显卡的计算能力为6.1,是支持CUDA版本的。

3. 查看tensorflow版本对应的CUDA和cudnn版本

tensorflow.png

在此我们选择Tensorflow-gpu 1.12.0对应的CUDA 9.0和cudnn 7.0版本

4. 安装CUDA 9.0

官网下载链接: https://developer.nvidia.com/cuda-toolkit-archive

cuda.png

安装完成后进行测试,在cmd中输入:

nvcc-V
nvcc.png

表示安装成功

windows10下使用nvidia-smi查看GPU使用情况:

nvidia-smi所在的位置为:C:\Program Files\NVIDIA Corporation\NVSMI,为了方便我们可以把此路径添加到path中。


nvidia-smi.png

看到显卡信息表示成功啦

4. cudnn库下载

官网下载链接:https://developer.nvidia.com/cudnn

cudnn.png

下载这个安装包需要注册并且填一堆问卷,下下来以后把相关包不用安装,直接拷到cuda路径对应的文件夹下面就行


cudnn1.png

cudnn3.png

TensorFlow-gpu配置流程

1. 安装TensorFlow

打开cmd,进入新建的tensorflow环境中

activate tensorflow

在cmd中使用pip安装输入以下指令:

pip install tensorflow-gpu=1.12.0

也可使用conda安装,有时候会出错……:

conda install tensorflow-gpu

2. 测试TensorFlow

测试TensorFlow是否安装成功,在cmd中输入python,进入Python编辑环境,输入以下指令

import tensorflow as tf
a = tf.constant(1.0)
b = tf.constant(2.0)
c = a + b
with tf.Session() as sess:
    print(sess.run(c))
tf_test.png

安装成功啦。

keras配置流程

1. 安装keras

使用conda安装:

conda install keras

2. 测试keras

import tensorflow as tf
import keras

测试结果:


keras.png

PyTorch-gpu配置流程

1. 安装PyTorch

同上使用conda创建一个pytorch的环境:

conda create -n pytorch python=3.6 ipykernel

进入pytorch环境:

activate pytorch

ptoch官网下载:https://pytorch.org/get-started/locally/

使用conda安装pytorch:

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

2. 测试PyTorch

测试CUDA与cuDNN是否工作正常:

# CUDA TEST
import torch
x = torch.Tensor([1.0])
xx = x.cuda()
print(xx)

# CUDNN TEST
from torch.backends import cudnn
print(cudnn.is_acceptable(xx))
pycharm2.png

安装成功啦~

PyCharm配置流程

1. 安装PyCharm

官网下载:https://www.jetbrains.com/pycharm/download/#section=windows

社区版免费,专业版收费,入门学习选择社区版即可

2. 配置PyCharm与TensorFlow

打开pycahrm,新建项目。


image

出现如下界面:


image

选择运行环境。选中Existing Interpreter,点击右边设置按钮,选择Add Local


image

点击Conda Enviroment,选择环境


image

进入Anaconda安装路径,选择envs文件夹,里面有建立的环境,选择之前建立的tensorflow环境中的python.exe,本机的路径为:D:\Anaconda3\envs\tensorflow\python.exe


pycharm.png

创建test.py文件测试tensorflow:

import tensorflow as tf
a = tf.constant(1.0)
b = tf.constant(2.0)
c = a + b
with tf.Session() as sess:
    print(sess.run(c))

输出结果:


pycharm1.png

成功配置tensorflow和pycharm啦~

3. 配置PyCharm与Pytorch

配置pytorch和pycharm与以上类似,就是把Python路径:由D:\Anaconda3\envs\tensorflow\python.exe更换为D:\Anaconda3\envs\pytorch\python.exe即可。

创建test.py文件测试pytorch:

# CUDA TEST
import torch
x = torch.Tensor([1.0])
xx = x.cuda()
print(xx)

# CUDNN TEST
from torch.backends import cudnn
print(cudnn.is_acceptable(xx))

输出结果:


pycharm3.png

同样配置成功了~

参考链接

  1. Win10下安装Anaconda+PyTorch+pycharm https://zhuanlan.zhihu.com/p/35255076
  2. Windows 安装 Anaconda3 详细过程 https://blog.csdn.net/u012318074/article/details/77075209
  3. 修改Jupyter Notebok的默认工作路径 https://zhuanlan.zhihu.com/p/48962153
  4. WIN10安装TENSORFLOW(GPU版本)详解(超详细,从零开始) https://zhuanlan.zhihu.com/p/37086409
  5. Anaconda使用笔记 https://zhuanlan.zhihu.com/p/57612562
  6. win10+cuda8.0+cudnn+Tensorflow(GPU)安装 https://blog.csdn.net/scythe666/article/details/78592756
  7. Win10 TensorFlow(gpu)安装详解 https://blog.csdn.net/sb19931201/article/details/53648615
  8. 如何查看电脑的显卡型号 http://xinzhi.wenda.so.com/a/1542603542209428
  9. Win10下安装Anaconda+PyTorch+pycharm https://zhuanlan.zhihu.com/p/35255076
  10. PyTorch在64位Windows下的Conda包 https://zhuanlan.zhihu.com/p/26871672
  11. Pycharm简易使用教程 https://zhuanlan.zhihu.com/p/52470112
  12. Deep-Learning-From-Zero-To-One https://github.com/chaowentao/Deep-Learning-From-Zero-To-One
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,080评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,422评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,630评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,554评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,662评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,856评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,014评论 3 408
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,752评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,212评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,541评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,687评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,347评论 4 331
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,973评论 3 315
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,777评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,006评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,406评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,576评论 2 349