WIN10+GTX1650TI显卡安装pytorch教程

2024-07-11 00:34| 来源: 网络整理| 查看: 265

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

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

nvidia-smi

如下图： å¨è¿éæå¥å¾çæè¿°

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

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

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

anaconda中新建环境 conda create --name pytorchgpu python=3.8（看你装的python是什么版本就写什么）

启动新环境

activate pytorchgpu

进入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

【本文地址】

今日新闻