RTX 5090 安装 pytorch3d

1. 显式指定 RTX 5090 的 Blackwell 架构 (sm_120),也可带上通用架构

export TORCH_CUDA_ARCH_LIST="12.0"

export FORCE_CUDA=1

2. 指定单卡并使用 --no-build-isolation 进行安装

CUDA_VISIBLE_DEVICES=0 pip install -e . --no-build-isolation

相关推荐
金伟API10246 小时前
如何从零打造一个极简的DeepSeek-R1大模型
人工智能·ai
江畔柳前堤6 小时前
大语言模型分布式训练:从并行策略到万卡工程的系统梳理
人工智能·分布式·深度学习·算法·目标检测·机器学习·语言模型
Shockang6 小时前
智能体环路工程实战
人工智能
美摄科技6 小时前
美摄美颜特效SDK Skill:以AI赋能智能视音频新视界
人工智能
Web3_Daisy7 小时前
Pump.fun 与 FOMO 竞争背后的 Meme 市场变局
大数据·人工智能·金融·web3·区块链
AI创界者7 小时前
MiniMax-H3 本地一键部署整合包:8G 显存玩转文图生视频、视频参考、角色替换与超分补帧全流程
人工智能·深度学习
江畔柳前堤7 小时前
HBM:大语言模型时代的「算力血液」——从内存墙到带宽革命的深度拆解
服务器·人工智能·windows·目标检测·语言模型·自然语言处理·软件工程
美摄科技7 小时前
视频一键成片SDK Skill:AI智能分析与语义理解
人工智能
北斗落凡尘8 小时前
LangGraph 入门实战(2)
python·langchain
fthux8 小时前
装闭 RenoPit 源码解析(05):FastAPI与Celery如何执行AI装修分析
人工智能·ai·开源·github·open source·renopit