有英伟达新卡的安装

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

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

其中cu118是cuda的版本

 

官网

https://download.pytorch.org/whl/

posted @ 2024-04-14 18:08  强仔必胜  阅读(3)  评论(0)  编辑  收藏  举报