PyTorch环境的配置和安装
记录一下PyTorch深度学习首先是annaconda的安装,进入www.anaconda.com网站下载软件。在一个项目中,可能会用到pytorch0.5版本、pytorch0.8版本,这就让我们在使用0.8版本的时候就要去卸载0.5的版本再重新安装pytorch0.8,这样做无疑是费时费力。为了解决这个问题,我们可以使用conda指令来创建pytorch的环境,实现在使用0.8版本的时候调用0
首先是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:刚刚开始学习,处于摸索阶段,肯定有很多不足,欢迎指正。
更多推荐
所有评论(0)