最新在Anaconda环境下安装pytorch以及cuda!!!

首先,去Anaconda官网下载并安装Anaconda(这里如果C盘不是很满的话,建议装在C盘),这里推荐去它的中文网(也可以去它的官网) (Anaconda安装的时候提示打的√尽量都√上,选Just me)

Anaconda 中文网

安装完毕之后点开始,找到 Anaconda Prompt 并打开

然后使用以下命令在Anaconda下创建虚拟环境:

conda create -n pytorch-g python=3.6(-n后面的为自定义环境名,笔者可自行定义,这里我们安装的pytorch的gpu版本,所以命名为pytorch-g,python=3.6为python版本)

创建成功后使用一下命令激活环境:

conda activate pytorch-g

可以看见括号里的base环境变为pytorch-g

然后:

pip list

只有上图所示的几个包

接下来,安装Pytorch以及cuda(这里我推荐直接去Pytorch官网安装,因为最新的cuda11.6和11.7版本直接包含在Pytorch里面,这里我推荐安装它的cuda=11.6版本)

首先,进入Pytorch官网:

PyTorch 

 

在这里windows系统我推荐使用conda命令下载,直接在刚创建的环境下输入以下命令: 

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

然后静静等待吧,有点点漫长!!!

完毕之后输入:

conda list

 发现该有的包都有了,往下拉还有torch等等就说明成功了,也可以再试试下面的命令:

conda install cudatoolkit=11.3

发现包已经都被安装上了,这就是新版本的方便之处!

 

 最后一步,验证是否成功:

输入:

python

import torch

torch.cuda.is_available()

若返回True,则说明成功啦!!!

 

 

 

  • 2
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值