深度学习1650ti在win10安装pytorch复盘 |
您所在的位置:网站首页 › 有1650ti显卡嘛 › 深度学习1650ti在win10安装pytorch复盘 |
深度学习1650ti在win10安装pytorch复盘
前言1. 安装anaconda2. 检查更新显卡驱动3. 根据pytorch选择CUDA版本4. 安装CUDA5. 安装cuDNN6. 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版本。 直接去官网下载。 Nvidia Geforce Experience更新显卡驱动为最新版本(24.03.17的1650ti驱动为551.76) 在pytorch官网中,可以结合现有的环境条件,查看可选的CUDA版本。注意最新pytorch版本需要python 3.8以上版本的python环境支持。个人选择了11.8版本的CUDA。 在CUDA Toolkit网站,寻找对应希望的CUDA版本。点击需要的11.8区域。 在cuDNN网站中寻找对应CUDA的最新版本。如下图所示。 Anaconda Prompt 激活步骤1中的新建环境,复制步骤2中的pytorch官网生成的命令,执行。使用魔法感觉是最稳妥的,不会出现什么换源之后默认下载cpu版本啥的稀奇古怪问题。
祝各位安装顺利~!时代潮流滚滚向前,个人也开始踏上学习AI之路。期待学习成果能用在robot人的工作上。 |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |