目的
本文主要记录自己安装pytorch过程。
环境:win10 pycharm
显卡:GTX1050Ti
过程记录
1、确认pytorch版本
打开pytorch官网:https://pytorch.org/

选择stable 2.3.0 版本,CUDA有11.8及12.1,我们看看GTX1050Ti支持哪个
在cmd里输入 nvidia-smi,可以看到当前支持12.2

所以我们这里选择pytorch-cuda12.1的版本。
复制对应的命令,在pycharm terminal里输入:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org


396

被折叠的 条评论
为什么被折叠?



