首页 最新 热门 推荐

  • 首页
  • 最新
  • 热门
  • 推荐

【配置环境】RTX3050安装pytorch(安装CUDA11.3版本)

  • 23-09-20 00:03
  • 4460
  • 7326
blog.csdn.net

目录

  • 参考链接
  • 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官网:http://iyenn.com/index/link?url=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环境可用。

文章知识点与官方知识档案匹配,可进一步学习相关知识
Python入门技能树人工智能深度学习331497 人正在系统学习中
注:本文转载自blog.csdn.net的孟孟单单的文章"https://blog.csdn.net/LWD19981223/article/details/126306276"。版权归原作者所有,此博客不拥有其著作权,亦不承担相应法律责任。如有侵权,请联系我们删除。
复制链接
复制链接
相关推荐
发表评论
登录后才能发表评论和回复 注册

/ 登录

评论记录:

未查询到任何数据!
回复评论:

分类栏目

后端 (14832) 前端 (14280) 移动开发 (3760) 编程语言 (3851) Java (3904) Python (3298) 人工智能 (10119) AIGC (2810) 大数据 (3499) 数据库 (3945) 数据结构与算法 (3757) 音视频 (2669) 云原生 (3145) 云平台 (2965) 前沿技术 (2993) 开源 (2160) 小程序 (2860) 运维 (2533) 服务器 (2698) 操作系统 (2325) 硬件开发 (2492) 嵌入式 (2955) 微软技术 (2769) 软件工程 (2056) 测试 (2865) 网络空间安全 (2948) 网络与通信 (2797) 用户体验设计 (2592) 学习和成长 (2593) 搜索 (2744) 开发工具 (7108) 游戏 (2829) HarmonyOS (2935) 区块链 (2782) 数学 (3112) 3C硬件 (2759) 资讯 (2909) Android (4709) iOS (1850) 代码人生 (3043) 阅读 (2841)

热门文章

101
推荐
关于我们 隐私政策 免责声明 联系我们
Copyright © 2020-2025 蚁人论坛 (iYenn.com) All Rights Reserved.
Scroll to Top