安装Anaconda,更新Pytorch到最新版本1.5。

3
如何使用Anaconda通过终端或导航器从1.4更新到1.5的PyTorch?
使用 conda update --all 更新Anaconda会更新一些包,但不包括 PyTorch。最初,我通过运行 conda install -c pytorch pytorch 安装了 PyTorch。
在PyTorch Github页面上,有以下命令:conda install -c pytorch magma-cuda90 # 或[magma-cuda92 | magma-cuda100 | magma-cuda101 ](根据您的cuda版本而定),但我想知道已安装版本和此版本之间是否会发生任何冲突。
谢谢!
1个回答

4

截至2020年4月6日,PyTorch的最新稳定版本仍为1.4,您可以在这里看到。

因此,如果您想使用conda安装夜间版本(正在追踪1.5),您可以按照官方说明进行操作:

  • Cuda 10.1:
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch-nightly -c defaults -c conda-forge
  • Cuda 9.2:
conda install pytorch torchvision cudatoolkit=9.2 -c pytorch-nightly -c defaults -c conda-forge -c numba/label/dev

或者仅限于 CPU 版本:

conda install pytorch torchvision cpuonly -c pytorch-nightly -c defaults -c conda-forge

或者,您可以等待1.5成为一个稳定版(目前我们处于发布候选版本2),然后像以前一样更新pytorch软件包。


请注意:

PyTorch 1.4是最后一个支持Python 2的版本

因此,如果您正在升级到PyTorch 1.5,则要告别Python 2(耶!!)。


谢谢!!! PyTorch 1.4存在的问题是设置卷积核大小或膨胀不够稳健。而在PyTorch 1.5中,它是稳健的,即如果最终(经过膨胀后)的卷积核大小大于输入图像,则自动应用填充到输入图像。这是通过检查两个版本中的相关代码得出的结论。留在1.4中需要更新有问题的代码。你对此有什么看法? - vpap
我在许多项目中需要时使用了夜间构建。我没有特别检查这些拉取请求,但如果它们对你有帮助,请使用 :) 如果这个答案对你有帮助,请不要忘记点赞。再见 Python 2 :) - Berriel
@vpap 只是为了补充一下背景,下一个发布版本(至少在2.x之前)应该提供向后兼容性(只有下一个“主要版本”会破坏先前的行为,虽然有一些例外情况,请参见 优化器和调度程序答案,因此您正在获得的更改应该会改善您的用户体验。如果不是这种情况,并且发生了严重的回归,您应该在官方PyTorch repo上提出问题。 - Szymon Maszke

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接