安装和配置Pytorch和cuda

请参考:https://www.cnblogs.com/tryhardwy/p/14659131.html

注意:直接粘贴

pip install torch==1.8.1+cu102 torchvision==0.9.0+cu102 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html,会提示找不到torchvision==0.9.0+cu102 导致安装失败!

正确的安装语句是:

pip install torch==1.8.1+cu102 torchvision==0.9.1+cu102 -f https://download.pytorch.org/whl/torch_stable.html

-------------文章已结束~感谢您的阅读-------------
穷且益坚,不堕青云之志。