在Anaconda环境下快速安装pytorch和CUDA!
首先,去Anaconda官网下载并安装Anaconda(这里如果C盘不是很满的话,建议装在C盘),这里推荐去它的中文网(也可以去它的官网) (Anaconda安装的时候提示打的√尽量都√上,选Just me)
安装完毕之后点开始,找到 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官网:
在这里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,则说明成功啦!!!