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可能时间要长一点,之后就很快了。

相关推荐
渡我白衣19 小时前
未来的 AI 操作系统(二)——世界即界面:自然语言成为新的人机交互协议
人工智能·语言模型·人机交互
道之极万物灭19 小时前
Python操作word实战
开发语言·python·word
墨利昂19 小时前
词向量:自然语言处理技术体系的核心基石
人工智能·自然语言处理
格林威19 小时前
可见光工业相机半导体制造领域中的应用
图像处理·人工智能·数码相机·计算机视觉·视觉检测·制造·工业相机
星期天要睡觉19 小时前
计算机视觉(opencv)——基于 MediaPipe 人体姿态检测
人工智能·opencv·计算机视觉
Python私教19 小时前
DRF:Django REST Framework框架介绍
后端·python·django
资讯全球19 小时前
2025机器人自动化打磨抛光设备及汽车零件打磨新技术10月应用解析
人工智能·机器人·自动化
数智前线19 小时前
京东零售的AI野心:为每个商家打造自己的“AI战队”
人工智能
Cl_rown去掉l变成C19 小时前
第N7周打卡:调用Gensim库训练Word2Vec模型
人工智能·自然语言处理·word2vec
腾讯云开发者20 小时前
腾讯云TVP走进美的,共探智能制造新范式
人工智能