网站建设的重要性网站建设 公众号
深度学习1650ti在win10安装pytorch复盘
- 前言
- 1. 安装anaconda
- 2. 检查更新显卡驱动
- 3. 根据pytorch选择CUDA版本
- 4. 安装CUDA
- 5. 安装cuDNN
- 6. conda安装pytorch
- 结语
前言
建议有条件的，可以在安装过程中，开启梯子。例如cuDNN安装时登录 or 注册，会加快不少。业余学习，用的21年入手的笔记本，显卡为GTX1650ti，win10系统上已成功安装pytorch2.2.1。
省流版：仅针对1650ti。先用Nvidia Geforce Experience更新显卡驱动为最新版本(24.03.17的1650ti驱动为551.76，对应的CUDA支持最高版本12.4)。pytorch2.2.1对应的python3.8——CUDA版本可选为11.8和12.1。个人选择的CUDA11.8安装。之后cuDNN版本选择与CUDA对应的8.9.7.29版本。
1. 安装anaconda
直接去官网下载。
个人使用的win10系统，就点击框选部分下载安装即可。可以大胆地认准官网提供的最新版本下载。对应捆绑的python版本无需关心，例如图片中的python 3.11。安装好之后，可以在Anaconda Prompt中，使用conda create -n yourname-xxx python=3.8,创建新的运行环境。指令中的yourname-xxx可以替换成任意字符命名。我这边用的py3.8。
2. 检查更新显卡驱动
Nvidia Geforce Experience更新显卡驱动为最新版本(24.03.17的1650ti驱动为551.76)
之后在Anaconda Prompt中键入指令nvidia-smi，即可查看对应驱动支持的CUDA最高版本。（最高 * 3，如图中的12.4以下版本皆可安装）
若未安装anaconda，也可以在任务栏的Nvidia 控制面板——帮助——系统信息——组件中查看支持的最高CUDA版本。
3. 根据pytorch选择CUDA版本
在pytorch官网中，可以结合现有的环境条件，查看可选的CUDA版本。注意最新pytorch版本需要python 3.8以上版本的python环境支持。个人选择了11.8版本的CUDA。
4. 安装CUDA
在CUDA Toolkit网站，寻找对应希望的CUDA版本。点击需要的11.8区域。
建议此处加上魔法，不然跳转后显示完整较慢。附魔后的页面，结合本人环境，勾选如下。
下载好之后，双击——安装选项选择 自定义即可。选择的组件，末尾几个有些可以取消勾选。取消原则为当前版本 比 新版本 高，那么对应的组件就可以取消勾选安装，如下图所示。
之后一路安装即可。安装结束后win + R，再cmd打开终端，输入nvcc -V指令(注意 V是大写的)，若安装成功，即可看到CUDA的版本信息。
5. 安装cuDNN
在cuDNN网站中寻找对应CUDA的最新版本。如下图所示。
点开后，即可在展开内容中选择需要的版本。
注意：这个页面下载需要登陆Nvidia账号。若未注册过，需要填写信息注册，建议也使用魔法，附魔后页面跳转显示速度更快。
压缩包下载后，解压，将这三个文件夹复制拷贝至CUDA的安装根目录中。
6. conda安装pytorch
Anaconda Prompt 激活步骤1中的新建环境，复制步骤2中的pytorch官网生成的命令，执行。使用魔法感觉是最稳妥的，不会出现什么换源之后默认下载cpu版本啥的稀奇古怪问题。
安装好之后，继续在虚拟环境中，开启python，依次输入
import torch
print(torch.__version__)
torch.cuda.is_available()
顺利显示版本号2.2.1和True后，说明安装成功。
在pycharm中，也可以验证一下。
结语
祝各位安装顺利~！时代潮流滚滚向前，个人也开始踏上学习AI之路。期待学习成果能用在robot人的工作上。