首先是annaconda的安装,进入www.anaconda.com网站下载软件。

在一个项目中,可能会用到pytorch0.5版本、pytorch0.8版本,这就让我们在使用0.8版本的时候就要去卸载0.5的版本再重新安装pytorch0.8,这样做无疑是费时费力。为了解决这个问题,我们可以使用conda指令来创建pytorch的环境,实现在使用0.8版本的时候调用0.8的版本环境,要使用0.5版本再重新调用pytorch0.5版本环境。

图1

图1显示的是使用conda创建环境成功。后面它会提示创建这个环境需要安装这些包,可以选择是否安装。

图2

图2就是选择安装这些包后的结果。可以看见要想激活这个环境可以输入conda activate pytorch指令,激活后,环境就从一开始的base变成了我们创建的pytorch。接着输入pip list指令可以查看这个环境中有哪些工具包,我们会发现package中没有pytorch,所以我们需要去下载安装pytorch。

图3 

进入https://www.geforce.cn/hardware/technology/cuda/supported-gpus的NVIDIA网站可以查看支持的GPU,进入https://pytorch.org/网站,选择合适的版本下载。在anaconda中输入网站的指令conda install pytorch torchvision torchaudio cpuonly -c pytorch,anaconda中会提示是否下载,选择yes开始下载pytorch。

图4 

输入pip list指令可以查看这个环境中有没有pytorch,可以看到pytorch已经安装成功。

图5 

输入import torch/torch.cuda.is_available() 指令,如果显示ture则表示已经完成安装。(如果电脑是AMD处理器,会显示false,但是也不影响下面的学习)

 ps:刚刚开始学习,处于摸索阶段,肯定有很多不足,欢迎指正。

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐