pytorch3d+pytorch1.10+MinkowskiEngine安装

1、配置pytorch1.10+cuda11.0

pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu111/torch_stable.html

2、配置 MinkowskiEngine库

不按下面步骤,出现错误

1、下载MinkowskiEngine==0.5.4到本地

2、查看自己电脑的计算能力,要和库匹配

复制代码
import torch
print(f"GPU Name: {torch.cuda.get_device_name(0)}")
print(f"Compute Capability: {torch.cuda.get_device_properties(0).major}.{torch.cuda.get_device_properties(0).minor}")

3、安装MinkowskiEngine库

复制代码
export TORCH_CUDA_ARCH_LIST="7.0"
export CXX=c++; export CUDA_HOME=/usr/local/cuda-11.0; python setup.py install --blas=openblas --force_cuda

因为的计算能力是7.0,cuda=11.0

3、安装 pytorch3d

要自己下载安装包pytorch3d==0.7.4

解压:cd pytorch3d && pip install -e

相关推荐
Joseit1 小时前
AI应用生成平台:数据库、缓存与存储
数据库·人工智能·缓存
QYR_111 小时前
2025-2031年全球箱体与盒体搬运机器人行业全景报告(含市场规模、竞争格局及投资潜力)
人工智能·市场研究
Lxinccode1 小时前
python(44) : docker构建支持消费rocketmq的客户端
python·docker·rocketmq·importerror·not found·dynamic library·pyrocketmq
绵羊20232 小时前
ImmuCellAI 免疫浸润分析
linux·python
袁总6663 小时前
智能座舱问答
人工智能
MediaTea3 小时前
Python IDE:Spyder
开发语言·ide·python
Sirius Wu4 小时前
SFT/DPO/PPO/GRPO训练全解析
人工智能·深度学习·语言模型
B站_计算机毕业设计之家4 小时前
大数据 Python小说数据分析平台 小说网数据爬取分析系统 Django框架 requests爬虫 Echarts图表 17k小说网 (源码)✅
大数据·爬虫·python·机器学习·数据分析·可视化·小说
Learn Beyond Limits4 小时前
Clustering|聚类
人工智能·深度学习·神经网络·机器学习·ai·聚类·吴恩达
KKKlucifer4 小时前
IDC发布AI+政务、财政、应急三大市场空间与厂商份额报告
人工智能·百度·政务