Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm

[Toc]

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

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

  • 系统: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,输入

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

jupyter notebook --generate-config

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

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

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

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

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

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

找到快捷方式,右键打开属性,将“目标”最后面的 “%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. 查看计算机显卡型号:

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

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

点击设备管理器。

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

点击显示适配器。

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

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

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

例如本机显卡型号为GTX1060

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

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

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

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

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

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

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

4. 安装CUDA 9.0

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

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

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

nvcc-V

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

表示安装成功

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

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

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

看到显卡信息表示成功啦

4. cudnn库下载

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

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

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

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

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))

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

安装成功啦。

keras配置流程

1. 安装keras

使用conda安装:

conda install keras

2. 测试keras

import tensorflow as tf
import keras

测试结果:

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

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))

安装成功啦~

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

PyCharm配置流程

1. 安装PyCharm

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

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

2. 配置PyCharm与TensorFlow

打开pycahrm,新建项目。

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

出现如下界面:

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

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

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

点击Conda Enviroment,选择环境

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

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

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

创建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))

输出结果:

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

成功配置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))

输出结果:

《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》
《Win10下安装Anaconda+CUDA+TensorFlow+PyTorch+PyCharm》

同样配置成功了~

参考链接

  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
    原文作者:披着狼皮的羊
    原文地址: https://zhuanlan.zhihu.com/p/59031211
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞