pytorch3d
·
使用VSCode创建Python虚拟环境:
打开VSCode,确保已安装Python扩展。按下Ctrl+Shift+P(Windows/Linux)或Cmd+Shift+P(Mac)打开命令面板。
输入Python: Create Environment并选择该命令。选择Venv作为环境类型,然后选择要使用的Python解释器版本。
指定虚拟环境的目录位置,通常选择项目文件夹。VSCode会自动创建虚拟环境并激活它,终端会显示(venv)前缀。
通过终端手动创建虚拟环境
在VSCode中打开集成终端(`Ctrl+``)。导航到项目目录,运行以下命令:
python -m venv venv
这会在当前目录创建名为venv的虚拟环境文件夹。激活虚拟环境的命令因操作系统而异:
Windows:
.\venv\Scripts\activate
配置VSCode使用虚拟环境
创建虚拟环境后,按下Ctrl+Shift+P打开命令面板。输入Python: Select Interpreter,选择虚拟环境中的Python解释器(通常位于venv/Scripts/python或venv/bin/python)。
安装pytorch
pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2
设置环境变量
set PYTORCH3D_NO_NINJA=1
set DISTUTILS_USE_SDK=1
安装pytorch3d
git clone https://github.com/facebookresearch/pytorch3d.git
cd pytorch3d
pip install -e . --no-deps # 忽略可能存在的CUDA依赖
可能还需要安装:pip install plotly
验证安装
import torch
from pytorch3d.renderer import MeshRenderer
print(torch.__version__, MeshRenderer) # 应输出版本号和类信息
安装第三方预编译包:
pip install --extra-index-url https://miropsota.github.io/torch_packages_builder pytorch3d==0.7.6+pt2.0.1cpu
查看安装的库
pip list
更多推荐


所有评论(0)