安装pytorch在其官网中需要对应Compute Platform的CUDA版本进行安装。查看windows系统的CUDA如下：
键盘win+R 输入cmd打开控制台，输入如下指令即可查看版本。

nvidia-smi

在这里插入图片描述
然后在pytorch官网选择合适的版本进行下载即可。在这里插入图片描述

# windows # pytorch # 深度学习
Logo
2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

cover

AI嚼数据吐模块？初级开发者的创意别慌，老码农教你玩出“反卷Buff”

avatar 2048 AI社区
cover

当AI开始“偷吃”用户数据并拼装功能模块：初级开发者的脑洞保卫战与老码农的灵魂蘸料

avatar 2048 AI社区

CodeGenie的AI辅助调优让你问题定位效率大幅提升

https://github.com/irqk3x/d/issues/1415https://github.com/irqk3x/d/issues/1414https://github.com/irqk3x/d/issues/1413https://github.com/irqk3x/d/issues/1412https://github.com/irqk3x/d/issues/1411https

avatar 2048 AI社区