700字范文,内容丰富有趣,生活中的好帮手!
700字范文 > TensorFlow+pytorch+gpu+anaconda极简安装教程

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

时间:2021-11-09 07:26:20

相关推荐

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

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

anaconda下载

清华镜像:下载链接

附赠:百度网盘

链接:/s/1QEgPKrsDZcZWaT0kFVCc6w提取码:v1x1anaconda安装

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 tfhello = tf.constant('Hello world!')sess = tf.Session()print(sess.run(hello))

3.pytorch安装

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

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

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。