当前位置:   article > 正文

Install torch and torchvision_installing torch and torchvision

installing torch and torchvision

The official stsps for installation is as follows:

  1. pip3 install https://download.pytorch.org/whl/cu80/torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl
  2. pip3 install torchvision

However, torchvision is with version 0.3.0 and is not compatible with pytorch 1.0.1

So, we need to change the second sentence to pip3 install torchvision==0.2.1

 

 

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

闽ICP备14008679号