python 如何在Google colab中更改pytorch版本

cygmwpex  于 2022-12-10  发布在  Python
关注(0)|答案(1)|浏览(571)

我需要更改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+ cuda 120
更重要的是,当我试图

pip uninstall torch

colab告诉我你想卸载pytorch-1.0.0吗
它是如何发生的?

dgsult0t

dgsult0t1#

首先你得跑

!pip uninstall torch -y

-y用于跳过提示请求。这将卸载torch,大约需要5分钟。
然后,您必须

!pip install torch==1.0.0

最后

import torch
torch.__version__
# '1.0.0'

相关问题