win10系统上安装pytorch3D

首先感谢前人的经验文章,使我少走不少弯路。最近三维视觉的文章越来越多,CVPR2022的三维视觉文章已经有了很大的比重。而最近关于点云的文章越来越多的使用pytorch3D来直接训练,要复现这些文章,无疑pytorch3D更加方便。
参考这篇文章以及pytorch3D的指引,我们首先使用anaconda3新建一个虚拟环境:

  conda create -n pytorch3d python=3.8
  conda activate pytorch3d

然后安装对应版本的包:

conda install -c pytorch pytorch=1.7.1 torchvision=0.8.2 cudatoolkit=10.2

其中,cudatoolkit版本要和自己的cuda版本一致。
然后进入地址,下载1.10.0的cub版本,解压,在电脑的系统变量中创建 CUB_HOME,将其值设为 你的解压地址/cub-1.10.0

Windows 10环境下安装PyTorch 3D通常涉及几个步骤: 1. **Python和环境**: 确保已经安装了最新版本的Python (建议使用Python 3.7或更高),以及Anaconda或Miniconda等科学计算环境管理器。 2. **安装CUDA和cuDNN**: PyTorch 3D需要NVIDIA GPU的支持,首先下载安装CUDA Toolkit,然后通过Conda或pip安装cuDNN。例如: ``` conda install cudatoolkit=11.2 -c nvidia pip install cupy-cuda112 ``` 需要选择适合当前GPU的版本。 3. **安装PyTorch**: 使用以下命令安装PyTorch,指定`--cuda`选项以便在GPU上运行: ```bash pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cpu/torch_stable.html --extra-index-url https://download.pytorch.org/whl/cu112/torch_stable.html ``` 如果你希望安装包含3D功能的版本,可以尝试安装: ```bash pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html --extra-index-url https://download.pytorch.org/whl/nightly/cu112/torch_nightly.html extras=contrib ``` 4. **安装PyTorch 3D**: 如果官方库尚未提供PyTorch 3D版本,你可以尝试从GitHub克隆并安装: ```bash git clone https://github.com/pytorch/vision.git cd vision pip install . ``` 并确保查看其最新的文档以确认是否支持Windows和你的CUDA版本。 5. **设置环境变量**: 更新系统环境变量,特别是添加CUDA的路径到`PATH`中。 6. **测试安装**: 创建一个小脚本,导入`torchvision.ops`模块来检查安装是否成功,并进行简单的3D模型测试。
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值