You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pytorch/杭州日电量/pip3正确安装PyTorch GPU版本.txt

11 lines
462 B
Plaintext

# 首先卸载
pip3 uninstall torch
pip3 uninstall torchvision
pip3 uninstall torchaudio
# 然后安装
pip3 install torch==2.1.0+cu121 torchvision==0.16.0+cu121 torchaudio==2.1.0+cu121 -f https://download.pytorch.org/whl/cu121/torch_stable.html
# 可临时清华源加速
pip3 install torch==2.1.0+cu121 torchvision==0.16.0+cu121 torchaudio==2.1.0+cu121 -f https://download.pytorch.org/whl/cu121/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple/