cuda12.0 安装 pytorch

前两天买的y7000p到了,然后就要重新配下环境。

流程如下

  1. 首先下载miniconda ,我下的是python3.8
  2. 创建自己的自定义环境
  3. 检查自己的cuda版本,我的是cuda:12.0
  4. 然后再pytorch上找到对应cuda版本的进行下载,pip install或者conda install 都可以

1.


直达链接

2.

conda create --name you_env_name python=3.8

conda activate you_env_name

3.

在cmd里输入 nvidia-smi 查看对应的 cuda 版本

4.

这里我们下载11.8 因为一般来说cuda会向下兼容的,但是不要差太多,拿捏不准可以都试试看。

直达链接

5.

测试下载好的 torch 是否可以使用cuda

最后,测试完毕,没有问题,不过第一次 import torch可能时间要长一点,之后就很快了。

相关推荐
黎燃8 小时前
短视频平台内容推荐算法优化:从协同过滤到多模态深度学习
人工智能
TF男孩9 小时前
ARQ:一款低成本的消息队列,实现每秒万级吞吐
后端·python·消息队列
飞哥数智坊9 小时前
多次尝试用 CodeBuddy 做小程序,最终我放弃了
人工智能·ai编程
后端小肥肠10 小时前
别再眼馋 10w + 治愈漫画!Coze 工作流 3 分钟出成品,小白可学
人工智能·aigc·coze
唐某人丶13 小时前
教你如何用 JS 实现 Agent 系统(2)—— 开发 ReAct 版本的“深度搜索”
前端·人工智能·aigc
FIT2CLOUD飞致云13 小时前
九月月报丨MaxKB在不同规模医疗机构的应用进展汇报
人工智能·开源
阿里云大数据AI技术13 小时前
【新模型速递】PAI-Model Gallery云上一键部署Qwen3-Next系列模型
人工智能
袁庭新13 小时前
全球首位AI机器人部长,背负反腐重任
人工智能·aigc
机器之心14 小时前
谁说Scaling Law到头了?新研究:每一步的微小提升会带来指数级增长
人工智能·openai
该用户已不存在14 小时前
Mojo vs Python vs Rust: 2025年搞AI,该学哪个?
后端·python·rust