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

cygmwpex 于 2022-12-10 发布在 Python

关注(0)|答案(1)|浏览(572)

我需要更改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吗
它是如何发生的？

python

来源：https://stackoverflow.com/questions/68903158/how-to-change-the-pytorch-version-in-google-colab

1条答案

按热度按时间

dgsult0t1#

首先你得跑

!pip uninstall torch -y

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

!pip install torch==1.0.0

最后

import torch
torch.__version__
# '1.0.0'

赞(0）回复(0）举报 2022-12-10

我来回答

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

1条答案

相关问题

热门标签

最新问答