码农知识堂 - 1000bd
  •   Python
  •   PHP
  •   JS/TS
  •   JAVA
  •   C/C++
  •   C#
  •   GO
  •   Kotlin
  •   Swift
  • RTX3050安装pytorch(安装CUDA11.3版本)


    目录

    • 参考链接
    • 0 查询NVIDIA GPU算力(可跳过)
    • 1 创建虚拟环境
    • 2 在线安装GPU版本

    参考链接

    • 强力推荐👉Pytorch1.10安装记录(CUDA11.3)
    • RTX3050ti、3060等系列显卡正确安装cuda、cudnn,配置Pytorch深度学习环境(避免由于版本不适应导致重装)
    • B站教学视频:【包教包会】Anaconda+Cuda+Cudnn+TensorFlow+Pytorch+Pycharm+Win10深度学习环境配置
    • 配置环境的一些坑

    🙋‍一些bug的解决参考链接:

    • ERROR: pip‘s dependency resolver does not currently take into account all the packages 解决办法分享

    ❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀❀

    跟着第一篇博文的安装步骤如下:有些语句和图像都是直接粘贴过来的,还是大佬牛🐂!


    0 查询NVIDIA GPU算力(可跳过)

    通过如下链接可以查询到NVIDIA显卡对应的算力,如果你的GPU型号不在表中,可能GPU太老,或者算力太低。(我就不应该觉得RTX3050好!下面的表中没有RTX3050,肯定是算力低!可是我因为看到他的NVCUDA自己从11.4升级到了11.6,就巴巴地把之前装好地CUDA11.3给卸载了,想重新配置11.6高级点的,结果没成功😭)

    • https://developer.nvidia.com/cuda-gpus#collapseOne

    1 创建虚拟环境

    用管理员运行Anaconda Prompt!!!

    在安装之前,强烈建议使用conda创建一个新的虚拟环境,这样就不会对你之前的环境照成影响。这里以创建torch1.10为例:

    • 这里创建一个名为torch1.10的虚拟环境,python使用3.8的版本
    conda create -n torch1.10 python=3.8
    
    • 1
    • 创建完成后,激活虚拟环境(注:后续的操作都是在该虚拟环境下进行)
    conda activate torch1.10
    
    • 1

    2 在线安装GPU版本

    首先进入PyTorch官网:https://pytorch.org/

    在这里插入图片描述

    通过官方的提示可以看到,自己电脑的GPU是支持CUDA10.2和CUDA11.3和CUDA11.6的(但是RTX可能还是带不动,还是安装11.3比较稳当),这里以CUDA11.3为例。

    指明安装版本:

    这里直接基于刚刚创建好的虚拟环境进行安装(注意:由于使用conda创建的虚拟环境这里直接用pip不用pip3),官方默认安装了torch、torchvision以及torchaudio三个包,假如你不需要使用torchaudio这个包,就把torchaudio==0.10.0+cu113指令删掉:

    (我删了torchaudio的,因为那是用于语音领域的,只要有视觉的torchvision就行了,我还在最后面加上了一句-i https://pypi.douban.com/simple,可帮助加速下载)

    pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html -i https://pypi.douban.com/simple
    
    • 1

    通过以上指令会把Pytorch1.10以及需要的CUDA11.3全部安装好(不需要单独安装CUDA)。

    安装完成后,进入python环境,检查CUDA是否可用,调用torch.cuda.is_available()返回True说明CUDA环境可用。

  • 相关阅读:
    一个简单的go使用grpc的案例
    PHP:命名空间必知必会
    SAP Router 配置手册
    463. 岛屿的周长
    vue3通过ref获取子组件defineExpose的数据和方法
    sqlsugar批量插入百万数据
    神经网络(一)基本概念
    如何自学网络安全 才不会成为脚本小子?
    基于Web的美食分享平台的设计与实现——HTML+CSS+JavaScript水果介绍网页设计(橙子之家)
    来自北大算法课的Leetcode题解:85. 最大矩形
  • 原文地址:https://blog.csdn.net/LWD19981223/article/details/126306276
  • 最新文章
  • 攻防演习之三天拿下官网站群
    数据安全治理学习——前期安全规划和安全管理体系建设
    企业安全 | 企业内一次钓鱼演练准备过程
    内网渗透测试 | Kerberos协议及其部分攻击手法
    0day的产生 | 不懂代码的"代码审计"
    安装scrcpy-client模块av模块异常,环境问题解决方案
    leetcode hot100【LeetCode 279. 完全平方数】java实现
    OpenWrt下安装Mosquitto
    AnatoMask论文汇总
    【AI日记】24.11.01 LangChain、openai api和github copilot
  • 热门文章
  • 十款代码表白小特效 一个比一个浪漫 赶紧收藏起来吧!!!
    奉劝各位学弟学妹们,该打造你的技术影响力了!
    五年了,我在 CSDN 的两个一百万。
    Java俄罗斯方块,老程序员花了一个周末,连接中学年代!
    面试官都震惊,你这网络基础可以啊!
    你真的会用百度吗?我不信 — 那些不为人知的搜索引擎语法
    心情不好的时候,用 Python 画棵樱花树送给自己吧
    通宵一晚做出来的一款类似CS的第一人称射击游戏Demo!原来做游戏也不是很难,连憨憨学妹都学会了!
    13 万字 C 语言从入门到精通保姆级教程2021 年版
    10行代码集2000张美女图,Python爬虫120例,再上征途
Copyright © 2022 侵权请联系2656653265@qq.com    京ICP备2022015340号-1
正则表达式工具 cron表达式工具 密码生成工具

京公网安备 11010502049817号