说明
说明显卡驱动对应cuda的问题(未来补充)
安装anaconda3
anaconda是一个我见过最好用的python包的管理软件,并且有图形化界面,想装什么包点一下就装了,非常的方便。废话不多说开始装:
1.点击进入anaconda3官网下载页面,下图是官网下载页面截图
2.往下拉,拉到如图所示位置,点击windouw后点击python3.7version下面的Download进行下载
3.下载结束后,双击安装包进行安装,选中图中所选后一路next
4.这就安装好了anaconda,可以在开始菜单里看到有这些选项
安装cuda+cudnn+tensorflow-gpu
传统的cuda和cudnn的安装方法非常的复杂,但是我们现在可以使用anaconda在安装tensorflow-gpu时顺带就把cuda和cudnn安装了,且是适配的,根本不用研究什么哪个tensorflow版本对应那个cuda和cudnn,更不用去了解繁琐cuda和cudnn的安装方法,不得不说ananconda是真的强大。
1.点击图中的Anaconda Prompt,会进入一个命令行
2.在命令行中输入以下命令创建虚拟环境,其中tensorflow_gpu是我们创建的虚拟环境的名字,这个自己可以随意改变的,可以取自己想要的名字。python=3.6代表我们搭建的虚拟环境是用python3.6这个版本建立的,当然你需要使用python3.5或python2.7都是可以自己改的。注意:输入下一行命令后会让我们确认一下,输入y然后回车就好。
conda create -n tensorflow_gpu python=3.6
3.输入以下代码进入我们刚搭建好的虚拟环境,tensorflow_gpu是虚拟环境的名称,应该改写为你自己环境的名称,我这里是tensorflow_gpu:
conda activate tensorflow_gpu
4.输入以下代码安装一些常用的库,这里基本会安装大部分我们将用到的库,一句代码全部安好,anaconda非常强大!
conda install anaconda
5.输入以下代码安装gpu版tensorflow,可以在图中看到,在这一步anaconda就直接给我们安装好了cuda和cudnn,非常的方便。
conda install tensorflow-gpu
5.如果你没有gpu,想安装cpu版本,请输入以下代码(安装gpu版tensorflow请忽略该步):
conda install tensorflow
6.完成以上步骤tensorflow-gpu版(tensorflow-cpu版)就装好了,依次输入以下前两行命令,测试是否装好tensorflow,如果是tensorflow-gpu版本还需要输入最后一行。如果显示如图,那就说明已经装好了tensorflow且能使用gpu计算,可以开始尽情码代码了!!!
python
import tensorflow as tf
sess = tf.Session(config = tf.ConfigProto(log_device_placement=True))
安装pytorch
1.点击进入pytorch官网,往下拉看到如下图所示,前面三行分别选择:stable1.1;windows;conda;后面两行根据当时我们装tensorflow-gpu时装的cuda版本以及python的版本选择,然后输入Run this Command栏中的代码,例如我这里是
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
2.因为我们之前就已经安装好了cuda和cudnn所以pytorch就已经安装好了。接下来可以输入以下代码看pytorch是否安装成功,pytorch是否能使用gpu。如果显示如图,那就说明已经装好了pytorch且能使用gpu计算,可以开始尽情码代码了!!!
python
import torch
torch.cuda.is_available()