当前位置:   article > 正文

colab中安装torch0.4.1_torch==0.4.1 torchvision==0.2.1

torch==0.4.1 torchvision==0.2.1

colab中安装torch0.4.1

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

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

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

!python -V
  • 1

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
  • 1
  • 2

在这里插入图片描述

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

#我打印出来torch == 0.4.1
import torch
print(torch.__version__)
  • 1
  • 2
  • 3

**注意 : ** 这里报错没关系一个一个看,先看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
  • 1
  • 2

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

在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/212146?site
推荐阅读
相关标签
  

闽ICP备14008679号