3

如何通过终端或导航器使用 Anaconda 从 1.4 -> 1.5 更新 PyTorch?

更新 Anaconda 并conda update --all更新了一些包,而不是全部包,包括 PyTorch。

最初,我通过运行安装 PyTorchconda install -c pytorch pytorch

从 PyTorch Github 页面有命令

conda install -c pytorch magma-cuda90 # or [magma-cuda92 | magma-cuda100 | magma-cuda101 ] depending on your cuda version

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

谢谢

4

1 回答 1

4

PyTorch 最新的稳定版本(截至 2020 年 4 月 6 日)仍然是 1.4,您可以在此处看到。

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

  • 库达 10.1:
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch-nightly -c defaults -c conda-forge
  • 库达 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(耶!!)。

于 2020-04-07T00:20:00.893 回答