如何安装pytorch1.7.1 python代码
时间: 2024-06-10 14:07:06 浏览: 120
如何安装pytorch
可以使用以下命令安装pytorch1.7.1:
pip install torch==1.7.1+cpu torchvision==0.8.2+cpu -f https://download.pytorch.org/whl/torch_stable.html
如果你需要GPU版本,可以使用以下命令:
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/cu110/torch_stable.html
阅读全文