你不会找到的,更简单的GPU的TensorFlow和pytorch安装教程!过程过于详细,所以图比较多,但不要怕,没有哪一步是会复杂的!
- anaconda下载
清华镜像:下载链接
附赠:百度网盘
链接:https://pan.baidu.com/s/1QEgPKrsDZcZWaT0kFVCc6w 提取码:v1x1 - anaconda安装
- 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,将对应指令在终端内输入。
安装过程如果出问题,九成以上的原因是各个安装包版本矛盾,找到适合自己显卡型号的版本才能解决问题。