我之前的老方法:
pip install torch
结果发现,显卡永不起来
打开官网:
https://pytorch.org/get-started/locally/
熟悉的界面。选和咱最匹配的
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
听官方的,总没错吧:
需下载7个小时,哈哈。
国内镜像
pip install torch -i xxx
额,不要搞笑行吗?这还是不会用GPU啊
pip install torch torchvision torchaudio --index-url https://mirror.nju.edu.cn/pytorch/whl/cu126
跟刚才的命令差不多。
时间缩短到半小时左右。