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

相关推荐
Aevget20 小时前
界面控件DevExpress WPF v25.2新功能预览 - 聚焦AI功能提升
人工智能·wpf·界面控件·devexpress·ui开发·.net 10
大邳草民20 小时前
Django 的动态特性:从 Python 动态机制到框架设计思想
笔记·python·django
用户37215742613520 小时前
Python 裁剪 PDF 教程:轻松裁剪页面并导出为图片
python
F_D_Z20 小时前
扩散模型对齐:DMPO 让模型更懂人类偏好
人工智能·扩散模型·kl散度·双阶段训练·散度最小化偏好优化
ezl1fe20 小时前
第一篇:把任意 HTTP API 一键变成 Agent 工具
人工智能·后端·算法
算家计算20 小时前
OpenAI推出首款浏览器,能否撼动全球超30亿用户的Chrome?
人工智能·openai·资讯
文火冰糖的硅基工坊20 小时前
[人工智能-大模型-33]:模型层技术 - 大模型的神经网络架构
人工智能·神经网络·架构
mit6.82421 小时前
[Agent可视化] docs | go/rust/py混构 | Temporal编排 | WASI沙箱
python
文火冰糖的硅基工坊21 小时前
[人工智能-大模型-33]:模型层技术概览 - 大模型内部组成与层次调用关系
人工智能
蜉蝣之翼❉21 小时前
检测十字标 opencv python
python·opencv·计算机视觉