ptyorch安装

  1. 查看python、pytorch、cuda以及显卡对应的版本

  2. nvidia-smi查看显卡驱动,若报错则需要安装显卡驱动

    • 这里的CUDA-Version意思是等下要安装的CUDA的版本要<=13.1
  3. nvcc -V查看是否安装了cuda,若报错,还需要去安装cuda工具包,这一步要注意版本匹配
    本地下载之后进行安装,一路默认即可。

    完成之后再次用nvcc -V命令检查:

  4. 登录pytorch官网,进行选择对应的版本进行安装

  5. 检查是否安装成功。python ->>import torch ->> torch.is_available()

相关推荐
大鹏的NLP博客16 小时前
深度学习模型PyTorch与ONNX数值一致性校验通用技术报告
人工智能·pytorch·深度学习·onnx
FriendshipT17 小时前
Ultralytics:解读BNContrastiveHead模块
人工智能·pytorch·python·深度学习·目标检测
程序员羽痕17 小时前
基于深度学习的眼疾识别系统
人工智能·pytorch·深度学习·分类·django
pokemen邪12 天前
PyTorch KernelAgent 源码解读 ---(6)--- Composer
人工智能·pytorch·composer
FriendshipT3 天前
Ultralytics:解读SPPF模块
人工智能·pytorch·python·深度学习·目标检测
不负信仰3 天前
PyTorch KernelAgent 源码解读 ---(5)--- Dispatcher
人工智能·pytorch·python
OpenApi.cc3 天前
猫狗分类模型 GPU版
pytorch·深度学习·分类
拥抱太阳06163 天前
HarmonyOS 应用开发《掌上英语》第2篇:ArkTS 模块化开发:Index.ets 统一导出模式深度解析
pytorch·ubuntu·harmonyos
拥抱太阳06163 天前
HarmonyOS 应用开发《掌上英语》第5篇:依赖注入与全局状态:AppStorageV2 实战HarmonyOS 应用开发《掌上英语》第1篇
pytorch·华为·harmonyos
无糖冰可乐214 天前
安装wsl2,并链接Windows上的vscode运行项目
pytorch·vscode·python·pip