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

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

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


    附赠:百度网盘
    链接:https://pan.baidu.com/s/1QEgPKrsDZcZWaT0kFVCc6w 提取码:v1x1
  2. anaconda安装
  3. TensorFlow-gpu和pytorch的安装

在安装之前,我们需要知道自己的笔记本支持的cuda版本,打开NVIDIA控制面板,点击右下角系统信息,如下,可以看到。以下安装过程所使用的的各个安装包的版本,都跟此有关,如果支持cuda 10.0,直接忽略版本号,安装最新的就好。

①打开anaconda,创建新环境

②打开终端

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

1.TensorFlow-GPU安装

conda install -c aaronzs tensorflow-gpu==1.9

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,将对应指令在终端内输入。

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

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值