TensorFlow+pytorch+gpu+anaconda极简安装教程

你不会找到的,更简单的GPU的TensorFlow和pytorch安装教程!过程过于详细,所以图比较多,但不要怕,没有哪一步是会复杂的!

  1. anaconda下载
    清华镜像:下载链接

    TensorFlow+pytorch+gpu+anaconda极简安装教程_第1张图片
    附赠:百度网盘
    链接:https://pan.baidu.com/s/1QEgPKrsDZcZWaT0kFVCc6w 提取码:v1x1
  2. anaconda安装
    TensorFlow+pytorch+gpu+anaconda极简安装教程_第2张图片TensorFlow+pytorch+gpu+anaconda极简安装教程_第3张图片TensorFlow+pytorch+gpu+anaconda极简安装教程_第4张图片TensorFlow+pytorch+gpu+anaconda极简安装教程_第5张图片TensorFlow+pytorch+gpu+anaconda极简安装教程_第6张图片TensorFlow+pytorch+gpu+anaconda极简安装教程_第7张图片TensorFlow+pytorch+gpu+anaconda极简安装教程_第8张图片TensorFlow+pytorch+gpu+anaconda极简安装教程_第9张图片TensorFlow+pytorch+gpu+anaconda极简安装教程_第10张图片TensorFlow+pytorch+gpu+anaconda极简安装教程_第11张图片
  3. TensorFlow-gpu和pytorch的安装

在安装之前,我们需要知道自己的笔记本支持的cuda版本,打开NVIDIA控制面板,点击右下角系统信息,如下,可以看到。以下安装过程所使用的的各个安装包的版本,都跟此有关,如果支持cuda 10.0,直接忽略版本号,安装最新的就好。
TensorFlow+pytorch+gpu+anaconda极简安装教程_第12张图片
①打开anaconda,创建新环境

TensorFlow+pytorch+gpu+anaconda极简安装教程_第13张图片

②打开终端

TensorFlow+pytorch+gpu+anaconda极简安装教程_第14张图片

TensorFlow+pytorch+gpu+anaconda极简安装教程_第15张图片

③在终端依次输入下列代码

1.TensorFlow-GPU安装

conda install -c aaronzs tensorflow-gpu==1.9

TensorFlow+pytorch+gpu+anaconda极简安装教程_第16张图片TensorFlow+pytorch+gpu+anaconda极简安装教程_第17张图片

2.cudnn安装,NVIDIA cuDNN是用于深度神经网络的GPU加速库,依次分别输入下面两行代码

conda install -c anaconda cudatoolkit==9.0
conda install -c anaconda cudnn==7.3.1

至此,TensorFlow-GPU已经安装完成,可以用下面代码检查是否安装是否正常,如果对pytorch没有需求,则到这里,就可以结束了。

import tensorflow as tf
hello = tf.constant('Hello world!')
sess = tf.Session()
print(sess.run(hello))

3.pytorch安装

打开官网:pytorch官网地址,选择适合自己的版本,这里我选择cuda 9.0,将对应指令在终端内输入。

TensorFlow+pytorch+gpu+anaconda极简安装教程_第18张图片

TensorFlow+pytorch+gpu+anaconda极简安装教程_第19张图片

TensorFlow+pytorch+gpu+anaconda极简安装教程_第20张图片

安装过程如果出问题,九成以上的原因是各个安装包版本矛盾,找到适合自己显卡型号的版本才能解决问题。

你可能感兴趣的:(tensorflow,pytorch,深度学习,机器学习,anaconda)