如何在Google colab中更改pytorch版本

Question

我需要更改 google colab 中的 pytorch 版本，所以我安装了 anaconda

%%bash
MINICONDA_INSTALLER_SCRIPT=Miniconda3-4.5.4-Linux-x86_64.sh
MINICONDA_PREFIX=/usr/local
wget https://repo.continuum.io/miniconda/$MINICONDA_INSTALLER_SCRIPT
chmod +x $MINICONDA_INSTALLER_SCRIPT
./$MINICONDA_INSTALLER_SCRIPT -b -f -p $MINICONDA_PREFIX

import sys
_ = (sys.path
        .append("/usr/local/lib/python3.6/site-packages"))

进而

!conda install  pytorch==1.0.0 torchvision==0.2.1 cuda100 -c pytorch --yes

但是当我

import torch
torch.__version__

是1.9+cuda120

更重要的是，当我试图

pip uninstall torch

colab 告诉我你要卸载 pytorch-1.0.0

这是怎么发生的？

Answer 1

首先，你必须运行

!pip uninstall torch -y

用于-y跳过提示请求。这将卸载torch，大约需要 5 分钟。

那么，你必须

!pip install torch==1.0.0

最后

import torch
torch.__version__
# '1.0.0'