PyTorch环境的配置和安装

首先是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:刚刚开始学习,处于摸索阶段,肯定有很多不足,欢迎指正。

  • 4
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
PyTorch是一个流行的深度学习框架,配置安装PyTorch环境可以通过以下步骤完成: 1. 首先,你需要安装Anaconda,它是一个用于科学计算的Python发行版。你可以从Anaconda官方网站下载适用于你操作系统的安装程序,并按照指示进行安装。 2. 安装Anaconda后,打开终端或命令提示符,创建一个新的虚拟环境。你可以使用以下命令创建一个名为"pytorch"的虚拟环境: ``` conda create -n pytorch ``` 3. 激活虚拟环境,输入以下命令: ``` conda activate pytorch ``` 4. 接下来,你可以使用conda命令安装PyTorch和相关的依赖项。根据你的需求和系统,你可以选择安装CPU版的PyTorch或CUDA版的PyTorch。以下是两个选项: - 安装CPU版的PyTorch: ``` conda install pytorch torchvision cpuonly -c pytorch ``` - 安装CUDA版的PyTorch(需要支持CUDA的显卡): ``` conda install pytorch torchvision cudatoolkit=xxx -c pytorch ``` 请注意,将"xxx"替换为你的CUDA版本,例如"10.1"。 5. 安装完成后,你可以在Jupyter Notebook中使用PyTorch。首先,确保你已经安装了Jupyter Notebook。然后,在激活的虚拟环境中,运行以下命令来安装nb_conda插件: ``` conda install nb_conda ``` 进入你想要添加到Jupyter Notebook中的虚拟环境,并运行以下命令来安装ipykernel: ``` conda install ipykernel ``` 最后,输入以下命令启动Jupyter Notebook: ``` jupyter notebook ``` 以上是配置安装PyTorch环境的一般步骤。如果你遇到任何问题,可以参考引用和引用中提供的更详细的说明。另外,你还可以查阅PyTorch官方文档和社区论坛以获取更多的帮助和资源。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

晓亮.

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值