Windows 下安装 CUDA 和 Pytorch
下载cuda并安装 CUDA Toolkit 11.6 Update 2 Downloads | NVIDIA Developer
查看duda版本
用Anaconda新建一个python3.8的环境
然后去Start Locally | PyTorch官网下载cuda对应的gpu环境(建议使用pip)选择具体细节后,复制命令安装
建议直接复制下面的代码
pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
检测安装
验证
《动手学深度学习》 — 动手学深度学习 2.0.0-beta0 documentation
然后解压
运行jupyter notebook
到达路径
点击
使用nvidia-smi查看使用率
解决办法
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 微软正式发布.NET 10 Preview 1:开启下一代开发框架新篇章
· 没有源码,如何修改代码逻辑?
· PowerShell开发游戏 · 打蜜蜂
· 在鹅厂做java开发是什么体验
· WPF到Web的无缝过渡:英雄联盟客户端的OpenSilver迁移实战