ubuntu16.04 python3.5.2 cuda10.1安装pytorch
现在由于Ubuntu已经很老了,以及python3.5.2不怎么支持了,到pytorch的官网看到,最低版本要求是cuda10.2,python3.9
pytorch官网:https://pytorch.org/
所以需要安装先前的版本:本文安装的版本是torch1.4.0,torchvision==0.5.0
的版本。
通过使用官方提供的命令安装:
pip install torch==1.4.0 torchvision==0.5.0
安装并且python测试成功: