在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,则说明成功啦!!!

 

 

 

物联沃分享整理
物联沃-IOTWORD物联网 » 在Anaconda环境下快速安装pytorch和CUDA!

发表评论