如何在 google colab 中每晚升级到 pytorch？

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 稳定版本）。