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

救命!!

Answer 1

你使用了错误的包名，正如 pytorch 网站上提到的那样使用这个:

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

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