WIN10+GTX1650TI显卡安装pytorch教程

 

 

1.首先安装annaconda  这个教程很多没有必要多说

重要的是安装完成后，打开ananconda promote，输入

nvidia-smi

如下图：

 

作用是检查电脑所支持的CUDA版本，如果不用这一步很容易就报错。

2.安装cuda和cudnn 这个教程很多 具体可以看这个视频，很详细

https://www.bilibili.com/video/BV1Rz411e7eJ/?spm_id_from=333.788.videocard.0

3.

anaconda中新建环境

conda create --name pytorchgpu python=3.8（看你装的python是什么版本就写什么）

启动新环境

activate pytorchgpu

4.

进入pytorch官网选择对应的命令

https://pytorch.org/get-started/locally/

换源安装

去掉官网给出的指令 -c后面的所有内容

conda install pytorch torchvision cudatoolkit=10.2

细节可以参考：https://blog.csdn.net/qq_44868807/article/details/106379783?utm_medium=distribute.pc_relevant.none-task-blog-title-3&spm=1001.2101.3001.4242 

5.安装pycharm

这部分具体教程可以看这个：https://www.bilibili.com/video/BV1hE411t7RN?p=2