目的
本文主要记录自己安装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