Anaconda,将Pytorch更新到最新版本1.5

时间:2020-04-06 20:12:39

标签: python anaconda pytorch anaconda3

我如何通过终端或导航器使用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 # or [magma-cuda92 | magma-cuda100 | magma-cuda101 ] depending on your cuda version

但是我想知道已经安装的版本和这个版本之间是否会有任何冲突。

谢谢

1 个答案:

答案 0 :(得分:3)

PyTorch的最新稳定版本(截至2020年4月6日)仍为1.4,如您所见here

因此,如果您想使用conda安装夜间版本(按1.5步进行),则可以遵循official instructions

  • 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成为稳定版本(当前,我们位于release candidate 2中),然后按照其他方式更新pytorch软件包。


请注意:

PyTorch 1.4 is the last release that supports Python 2

因此,如果您要使用PyTorch 1.5,请告别Python 2(是的!)。