win10装anaconda3+cuda+cudnn+tensorflow-gpu+pytorch-gpu最详细教程

###说明
说明显卡驱动对应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()

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值