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

相关推荐
Robot侠2 小时前
极简LLM入门指南4
大数据·python·llm·prompt·提示工程
小a杰.2 小时前
Flutter 与 AI 深度集成指南:从基础实现到高级应用
人工智能·flutter
colorknight2 小时前
数据编织-异构数据存储的自动化治理
数据仓库·人工智能·数据治理·数据湖·数据科学·数据编织·自动化治理
Lun3866buzha2 小时前
篮球场景目标检测与定位_YOLO11-RFPN实现详解
人工智能·目标检测·计算机视觉
janefir2 小时前
LangChain框架下DirectoryLoader使用报错zipfile.BadZipFile
人工智能·langchain
等....2 小时前
Miniconda使用
开发语言·python
Java&Develop3 小时前
Aes加密 GCM java
java·开发语言·python
齐齐大魔王3 小时前
COCO 数据集
人工智能·机器学习
AI营销实验室4 小时前
原圈科技AI CRM系统赋能销售新未来,行业应用与创新点评
人工智能·科技
爱笑的眼睛114 小时前
超越MSE与交叉熵:深度解析损失函数的动态本质与高阶设计
java·人工智能·python·ai