colab中安装torch0.4.1

colab中安装torch0.4.1

1.whl列表查看torch历史版本找到0.4.1

https://download.pytorch.org/whl/torch_stable.html

2.查看当前colab中python版本号

!python -V

3.在历史版本中找到对应python的whl文件(我的是py3.7.x版本的)- - -就是cp37-cp37m这一块对应好

# 在这里用的是cu90下的.whl文件,别找错了有的是cpu下的
!pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp37-cp37m-linux_x86_64.whl

colab中安装torch0.4.1_第1张图片

这里提示下载成功了,然后可以打印看一下现在torch版本是否是0.4.1

#我打印出来torch == 0.4.1
import torch
print(torch.__version__)

**注意 : ** 这里报错没关系一个一个看,先看torchvision在网上找到这里应该用0.2.1版本的才可以与torch0.4.1对应

#使用这行代码进行安装
!pip install --index-url https://test.pypi.org/simple/ --extra-index-url https://pypi.org/simple/ torchvision==0.2.1.post2

colab中安装torch0.4.1_第2张图片

这一下就只剩一个包版本包不支持了,但是我没管它对于后续代码运行没带来错误只要import torchvision 成功就行,这个时候torch0.4.1就安装完成了(如果要用到fastal这个包就继续去找对应版本去下载就行了)

在这里插入图片描述

你可能感兴趣的:(pytorch,深度学习,python)