windows下cuda、cudnn以及pytorch的安装
一.在anaconda下配置cuda、cudnn以及pytorch环境
1.打开Anaconda Prompt,输入
conda create -n pytorch python=3.8 (这里环境名字命名为pytorch)
2.激活环境
activate pytorch
3.查看自己电脑显卡驱动版本,以及显卡驱动对应的最高cuda版本
在dos命令行输入
nvidia-smi
本机显卡驱动为511.79,如果不是最新版本可以先把驱动升级为最新版本,因为驱动对cuda的版本是向下兼容的,这样可以选择的cuda版本会更多
本机兼容cuda最高版本为11.6
4.进入pytorch官网,找到合适的cuda版本对应的pytorch,复制安装的代码到激活的pytorch环境中
在pytorch环境输入conda install pytorch torchvision torchaudio cudatoolkit=11.3 ,等待安装即可
二.在windows下配置cuda、cudnn以及pytorch环境
1.下载Python解释器
到Python官网下载Python解释器,本文下载Python3.8版本,下载下来后安装即可
2.根据一种的方法到NVIDIA官网下载对应cuda版本
3.到NVIDIA官网下载cuda版本对应的cudnn,并把文件复制到cuda制定安装目录中
4.配置pytorch
进入pytorch官网,如一所示,package选择pip,复制对应代码到dos窗口,如果下载出现时间过长,安装失败错误
可用pip --default-timeout=100 install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio===0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html命令
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 在鹅厂做java开发是什么体验
· 百万级群聊的设计实践
· WPF到Web的无缝过渡:英雄联盟客户端的OpenSilver迁移实战
· 永远不要相信用户的输入:从 SQL 注入攻防看输入验证的重要性
· 全网最简单!3分钟用满血DeepSeek R1开发一款AI智能客服,零代码轻松接入微信、公众号、小程