安装pytorch:选择对应版本

由于服务器网络问题,pip命令安装时速度非常慢,选择分别去网站:

下载两个文件:
torchvision-0.6.0+cu101-cp36-cp36m-linux_x86_64.whl
torch-1.5.0+cu101-cp36-cp36m-linux_x86_64.whl
下载完成后传到服务器,安装这两个文件:
pip3 install torch-1.5.0+cu101-cp36-cp36m-linux_x86_64.whl torchvision-0.6.0+cu101-cp36-cp36m-linux_x86_64.whl
测试:
In [1]: import torchvision
In [2]: import torch
In [3]: torch.cuda.is_available()
Out[3]: True
完成pytorch的安装。
安装pycocotools:
pip install cython
pip install -U 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
本文提供了一种解决服务器网络慢导致的PyTorch安装问题的方法,通过直接下载并安装特定版本的PyTorch和torchvision轮子文件,以及详细步骤指导如何安装pycocotools库。
1501

被折叠的 条评论
为什么被折叠?



