Win10+GTX1050Ti安装Pytorch

目的

本文主要记录自己安装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/whl/cu121

安装完成即可。

2、安装对应CUDA及cudnn
1）打开CUDA官网，https://developer.nvidia.com/cuda-toolkit-archive
2）下载并安装对应版本。
3）设置环境变量：

4）打开cudnn网址，https://developer.nvidia.com/cudnn
5）下载并安装版本。
如果需要注册账号，可以简单注册一个。

3、测试
pycharm里新建工程，新建文件。

import torch
torch.cuda.is_available()

print(torch.cuda.device_count())

输出结果：

参考

https://zhuanlan.zhihu.com/p/583845868