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

相关推荐
搬砖的小码农_Sky9 分钟前
AI:机器人行业发展现状
人工智能·机器人
深圳市快瞳科技有限公司22 分钟前
破解多宠管理难题,端侧AI重新定义宠物智能硬件
人工智能·智能硬件·宠物
276695829233 分钟前
tiktok 弹幕 逆向分析
java·python·tiktok·tiktok弹幕·tiktok弹幕逆向分析·a-bogus·x-gnarly
Blossom.11837 分钟前
用一张“冰裂纹”石墨烯薄膜,让被动散热也能做 AI 推理——基于亚波长裂纹等离激元的零功耗温度-逻辑门
人工智能·深度学习·神经网络·目标检测·机器学习·机器人·语音识别
cylat39 分钟前
Day59 经典时序预测模型3
人工智能·python·深度学习·神经网络
嘉恩督43 分钟前
视频人脸处理——人脸面部动作提取
python·音视频
WJ.Polar1 小时前
Python数据容器-集合set
开发语言·python
萤火虫儿飞飞1 小时前
从基础加热到智能生态跨越:艾芬达用创新重构行业价值边界!
大数据·人工智能·重构
aneasystone本尊1 小时前
学习 RAGFlow 的系统架构
人工智能
Codebee1 小时前
OneCode3.0低代码引擎核心技术:常用动作事件速查手册及注解驱动开发详解
人工智能·架构