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

相关推荐
ㄟ留恋さ寂寞几秒前
JavaScript中箭头函数在大括号省略时的隐式返回机制
jvm·数据库·python
WangN29 分钟前
【SONIC】Isaac Lab 系统入门指南
人工智能·python·机器人·自动驾驶·仿真
沫儿笙14 分钟前
库卡机器人二保焊混合气节气装置
网络·人工智能·机器人
SelectDB技术团队14 分钟前
强行拍平?全表扫描? AI Agent 动态 JSON 的观测分析
数据库·人工智能·json·apache doris
大模型最新论文速读15 分钟前
EvoLM:8B 模型自写评分标准,RL 后超越 GPT-4
人工智能·深度学习·算法·机器学习·自然语言处理
2501_9012005318 分钟前
Laravel 大批量数据填充时的内存泄漏与性能优化指南
jvm·数据库·python
晓山清24 分钟前
TCN时序卷积网络详解
网络·人工智能·cnn·时序卷积网络
weixin_3975780230 分钟前
本地git 的使用
人工智能
霸道流氓气质31 分钟前
Spring AI ChatMemory 对话记忆配置JDBC方式到Mysql数据库实战示例与原理讲解
数据库·人工智能·spring
搬砖的梦先生34 分钟前
Codex 全 AI 辅助测试操作指南
人工智能