python - 如何在 google colab 中升级到 pytorch-nightly？

Question

我遇到了问题，我该如何解决？我想在 colab 上运行 pytorch-nightly，因为里面有新的包，所以我的所有代码都在 pytorch-nightly 版本中，

我试图对此进行搜索并尝试了此代码，但即使在重新启动运行时它也无法正常工作

from os.path import exists
cuda_output = !ldconfig -p|grep cudart.so|sed -e 's/.*\.\([0-9]*\)\.\([0-9]*\)$/cu\1\2/'
accelerator = cuda_output[0] if exists('/dev/nvidia0') else 'cpu'
!pip install torchvision_nightly
!pip install torch_nightly -f https://download.pytorch.org/whl/nightly/{accelerator}/torch_nightly.html

帮助！！

标签： pythonpytorchgoogle-colaboratory

Answer 1

您使用了错误的包名称，如 pytorch 网站上所述，请使用：

!pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html -U

在这里，-U选项用于升级（因为 colab 上已经安装了 pytorch 稳定版本）。