windows10安装Anaconda,cuda和GPU版Pytorch

windows10安装Anaconda,cuda和GPU版Pytorch (简单详细)

1.Anaconda

安装anaconda
link
在这里插入图片描述

2.Cuda

如果要使用GPU版本的Pytorch则必须安装Cuda
link

![在这里插入图片描述](https://img-blog.csdnimg.cn/20201206000702697.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nL在这里插入图片描述

3.Pytorch

在官网找到适合自己的Pytorch版本
link
在这里插入图片描述

这时候打开anaconda prompt, 创建一个虚拟环境,名字随便起,我这里叫"torch_py38" (提醒自己用的是python3.8).

你在创建环境的时候也要注意自己使用的是什么版本的python,python版本,cuda版本和anaconda版本一定都要保持一致要不很容易有奇怪的bug…OvO’’

在这里插入图片描述

然后我们进入这个环境, 可以看到 base -> torch_py38
在这里插入图片描述

然后输入上面在pytorch官网找到的命令
我的版本是这个
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

在这里插入图片描述

装好之后测试一下:
在这里插入图片描述
如果第一个命令得到True就大功告成了。
如果第一个命令false但是底下的测试成功那么代表pytorch还是没有使用你的GPU,这时候就得倒回去检查是不是前面哪一步做漏了。

也有可能是出现了其他的bug,这里举例几个:

  1. 没有更新显卡驱动
    显卡驱动可以到NVDIA官网自己更新下载:link
    选择自己的型号就好
  2. python 版本不对,这边建议简单粗暴直接删除刚才创建的虚拟环境然后从创建虚拟环境那一步开始使用正确的python版本从做一次
  3. pytorch 版本不对,在刚才的虚拟环境里,输入 conda uninstall pytorch, 然后找到正确的版本命令重装pytorch。记得一定要先把之前的版本给卸载掉要不新的版本安装会出很多问题 (0.0

以上就是在windows 10 系统下安装 GPU版的Pytorch的全部过程了
(文中在最开始省略了安装python的步骤,
但是笔者认为没有必要讲解这一步,
如果您因为没做这一步而没法成功安装…)

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值