记录踩过的坑-PyTorch

天边一坨浮云2024-03-04 11:24

安装报错

按PyTorch官网给出的命令

复制代码

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

报错

复制代码

ERROR: Could not find a version that satisfies the requirement torch (from versions: none)
ERROR: No matching distribution found for torch

原因：当前python版本过低（我是3.6）

解决方法：用Anaconda重新建一个Python高版本环境（比如3.10）

我的是Windows系统，不支持nccl

在运行的py文件最开始加代码

复制代码

import os
os.environ["PL_TORCH_DISTRIBUTED_BACKEND"] = "gloo"

修改类似地方，把nccl改成gloo

复制代码

torch.distributed.init_process_group("nccl")