谷歌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'