首先准备好我们需要的东西pycharm,cuda,anaconda,python
我的电脑型号:以及相关配置
电脑型号:DELL G3---windows10
GPU:NVIDIA GeForce GTX 1650
pycharm,anaconda在网上随便找教程就可以下载:
这里我推荐在b占看:最详细的 Windows 下 PyTorch 入门深度学习环境安装与配置 CPU GPU 版 | 土堆教程
这里我着重讲一下cuda的下载
这个b站中的视频里面有cuda的下载教程分别对应了三种方法但是我用了前两种方法都不可以,所以我这里建议直接采用第三种简单,并且也不会花费过多的时间。
然后直接点击get started;进入页面后滑动到下侧这个页面这里就按照自己电脑的配置来进行相应的下载,一般下载stable版本就可以
这里就得查看好自己的gpu支持哪个版本的cuda;我们打开 NVIDIA Control Panel(直接在电脑左下角搜索栏搜索就可以看到);然后打开后点击系统信息
然后点击组件在3D设置里面就可以看到此硬件支持的cuda版本这里我的是12.7.33建议一般下载比你支持的版本低一点的这样会减少很多错误。
然后我们就知道自己的cuda版本了然后回到这个页面选择stable稳定版,然后选择自己的系统(我的是windows),下载包的方式(我这里选的是pip),cuda版本;最后在最下面有一段下载链接把它复制好,找到自己的迅雷(迅雷比一般的可以快两个多小时)
我这里选的是cuda12.4,打开迅雷后,新建任务然后把上图所示最后一行粘贴进去
然后把前面pip到url的全部删除然后就可以下载了
等下载完成之后打开文件夹找到下载好的whl文件,再打开anaconda powshell启动建好的虚拟环境jin,然后输入pip install 把下载好的文件拖进去如图所示,按下回车即可这里因为本地已经有了这个文件所以这次就下载的很快。
最后出现Successfully installed说明下载成功。这里就解决很多的问题了但是此时我们只下载了torch,还有torchvision torchaudio还没下载然后就在此页面继续输入
pip3 install torch torchvision torchaudio
这样就说明他开始下载了
同样出现Successfully installed说明pytorch-cuda下载成功。
但是我们得验证一下是否下载成功打开anaconda power shell激活自己的环境然后输入conda list即可如图所示说明安装cuda成功。