Win11安装gpu版pytorch完整教程

Win11系统/RTX30系列显卡——安装gpu版pytorch完整教程

1.电脑显卡:RTX 3060

Win11安装gpu版pytorch完整教程

; 2.预先安装 cuda11.4.3(支持win11的最低版本)

Win11安装gpu版pytorch完整教程
参考教程:https://blog.csdn.net/weixin_42818074/article/details/121143131

Win11安装gpu版pytorch完整教程
参考教程:https://blog.csdn.net/qq_37296487/article/details/83028394

3.安装Anaconda3 2021.11 和 Python 3.9.7

conda install pytorch torchvision torchaudio cudatoolkit=11.3(去掉-c后面的代码)
此时安装的时pytorch1.10.1(经测试,pytorch1.2.0+cudatoolkit11.3或pytorch1.2.0+cudatoolkit10.1都失败,可能时因为显卡3060不支持cuda11.1以下的版本&pytorch1.1不支持高版本cuda)
经过n此换源尝试(不同源的顺序很重要,决定能否install成功),最后默认源设置如下

Win11安装gpu版pytorch完整教程
测试pytorch-cuda是否安装成功,输出true则为安装成功
Win11安装gpu版pytorch完整教程
Win11安装gpu版pytorch完整教程

; 番外:版本检测

检查安装完的pytorch

  • cmd中输出:conda list
    查看 torch是否为cuda版本(有的源下载后为cpu版本,无法使用cuda)
    Win11安装gpu版pytorch完整教程
    Win11安装gpu版pytorch完整教程
  • anaconda中检查
    Win11安装gpu版pytorch完整教程
    Win11安装gpu版pytorch完整教程

Original: https://blog.csdn.net/weixin_43095105/article/details/124787403
Author: 湘潇一海
Title: Win11安装gpu版pytorch完整教程

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/709663/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球