linux 安装pytorch3d的坑

这篇具有很好参考价值的文章主要介绍了linux 安装pytorch3d的坑。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

事实上,只要按照官方文档的说明就可以完美安装。其中坑的地方在于conda的管理可能会导致下载的版本不符合你的要求(例如下载成了cpu版本、下载的cuda版本)而同样尝试使用源码编译以及其他方式下载库都会导致同样的问题,这里主要的原因是由于python的版本不对以及conda 版本不对。这里先简单总结一下本人成功安装的步骤,再简单阐述一下其中的坑,请务必完整阅读后再进行安装。

安装步骤

由于pytorch3d对于conda的要求非常刁钻,!!!这里十分建议重新创建一个虚拟环境!!!
官方链接
此外,目前测试python3.11还不可用,python3.8有点老(不太推荐)

conda create -n pytorch3d python=3.10
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
conda install -c fvcore -c iopath -c conda-forge fvcore iopath
conda install pytorch3d -c pytorch3d
#for test
pip install scikit-image matplotlib imageio plotly opencv-python

问题列表

1、conda问题

下载速度过慢/solving environment 一直转圈

下载速度过慢可以尝试更换源
清华源

vim ~/.condarc

清除原有内容,将下面内容复制进去。保存并退出。

channels:
  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  deepmodeling: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/

最后,使用如下命令

 conda clean -i

如果出现了solving environment转圈然后失败,可能是因为你之前就已经通过命令行的方式换过源了,这将会导致换源后需要搜索的目标过大,从而导致解析失败。解决方法与上述一致,更新过后的源将会是非常干净的清华源。
当然,一直转圈的原因也有可能是conda版本过旧,可以使用如下方法更新conda

更新conda

conda update -n base -c defaults conda

若失败,则尝试

conda update -n base -c defaults conda --repodata-fn=repodata.json

2、安装pytorch3d中重新又安装了一个pytorch的cpu版本

(1)我们强烈建议先检查一下conda版本是否过低,当确认无误后再查看本节。
(2)注意:请不要使用pip来安装pytorch环境
这是因为使用pip安装的pytorch环境将不会被conda安装的pytorch3d环境检测到,这样的话会导致其认为没有pytorch的合适版本而重新安装一次,这样将会导致你之前的pytorch环境完全崩溃
(3)请检查python版本是否过老或过新

3、cuda问题

有关cuda的一些简单概念参考
我们知道pytorch和pytorch都需要使用gpu版本。但是我们需要注意到pytorch系列使用到的cuda相关工具都会再conda安装时被自动嵌入到相关环境中(如果你仔细观察会发现conda下载时会下载一些带有cuda的包),老版本的pytorch中会下一个cudatoolkit,好像新版本中不需要了。因此,我们这里所参考的cuda版本实际上就是pytorch对应后面的版本,例如
linux 安装pytorch3d的坑,linux,pytorch,pytorch3d
而像外部环境中nvida-smi 和nvcc- V的版本这里都不用管
为了增加论证,这里列出我的这两个相应的显示结果
linux 安装pytorch3d的坑,linux,pytorch,pytorch3d
linux 安装pytorch3d的坑,linux,pytorch,pytorch3d
可以看出,我的这两个版本与我下载的pytorch-cuda版本均不一致,但不影响使用
by the way:下载pytorch的时候还是要保证其gpu版本尽量和nvidia-smi显示的版本尽量一致,否则会出现虽然正常检测但是运算速度极慢的情况。

测试

请在安装pytorch后测试如下代码,检测你的版本和gpu是否可用

import torch

print(torch.__version__)
print(torch.cuda.is_available())

请在安装pytorch3d后测试如下代码,检测你的pytorch3d是否可用
CMU 3dvision 作业1文章来源地址https://www.toymoban.com/news/detail-597436.html

到了这里,关于linux 安装pytorch3d的坑的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • Windows安装Pytorch3d

    安装Visual Studio 2019 【我记得必须是2017-2019之间的版本,我一开始用的是2022的版本就安装不了】网址 pytorch和pytorch3d、cuda和NVIDIA CUB版本需要相互对应 pytorch和pytorch3d版本对应关系如下:https://github.com/facebookresearch/pytorch3d/releases cuda和NVIDIA CUB版本对应关系如下:https://github.com/

    2024年02月09日
    浏览(31)
  • Pytorch3D安装全流程-亲自安装

    简单介绍 Pytorch3D是一个用于加速深度学习在处理3D相关信息时候的运算速率的库。深度学习有时会需要处理大量的3D数据,比如在人脸建模的时候。这时候使用这些库会大大帮助我们节省运算成本。 遇到问题 按照官方的安装方法会遇到不少问题:https://github.com/facebookresearch

    2024年01月17日
    浏览(28)
  • 安装pytorch3d最简单方法

    安装pytorch3d踩了很多坑,现将最简单的方法公布如下: PyTorch3D的目标是帮助加速深度学习和3D交叉点的研究。3D数据比2D图像更复杂,在从事Mesh R-CNN和C3DPO等项目时,我们遇到了一些挑战,包括3D数据表示、批处理和速度。我们开发了许多有用的算子和抽象,用于3D深度学习,

    2023年04月09日
    浏览(29)
  • Windows下Pytorch3d的安装方法

    一、问题: 近期在入手三维重建中,发现很多项目程序都需要Pytorch3d库,其官网定义为: PyTorch3D provides efficient, reusable components for 3D Computer Vision research with PyTorch. 可知Pytorch3d库可以提供多种应用于3D视觉处理的组件,其在ubuntu下安装较为简单,但win下稍繁琐,本贴是对个人安

    2024年02月02日
    浏览(34)
  • Windows GPU 下Pytorch3d的安装方法

    我的笔记本是windows GPU 安装Pytorch3d失败,废了九牛二虎之力才安装好,为了纪念我的坚持我分享给大家。 Pytorch3d在WIN下的安装 一、问题: 近期在入手三维重建中,发现很多项目程序都需要Pytorch3d库,其官网定义为: PyTorch3D provides efficient, reusable components for 3D Computer Vision r

    2024年02月03日
    浏览(49)
  • pytorch3d安装遇到的一些坑和解决过程

    最近为了安装pytorch3d,折腾了两天,足足两天! 要注意,安装pytorch3d之前,必须先安装pytorch torchvision Pytorch3d官方 是建议用conda install的方式来安装,也可以用pip来安装: pip install PyTorch torchvision -c pytorch-nightly 1.首先说一下MacOS M1 芯片安装情况,刚开始使用的是: 会安装完成

    2024年02月06日
    浏览(28)
  • Win10下pytorch3D安装方法,本人亲测可用

    本人的电脑配置如下: 目录 1.版本确认 2.下载cub 3.下载pytorch3d  4.下载VSC2019  5.编译pytorch3d源文件         这个步骤非常非常重要,因为不同的pytorch3d版本适配于不同的torch版本,你可以进到facebook/pytorch3d的官网查看不同的适配版本(这是0.7.1的),其余的你们可以自己查看

    2024年02月15日
    浏览(33)
  • Windows 11 安装 pytorch3d可能遇到的问题,以及最终成功安装使用的版本分享(使用RTX3070)

    Win10下pytorch3D安装方法   由于pytorch3d对于pytorch、CUDA、CUB的版本对应实在是过于严苛,所以我的建议是直接找别人成功安装的软件版本列表对着抄,pytorch啥的该重装重装,反正可以用虚拟环境,CUDA也是,一台电脑共存两个版本的CUDA也不是不行。唯一麻烦一点的可能就是G

    2024年02月20日
    浏览(32)
  • win下配置pytorch3d

    你可能觉得pytorch3d 0.6.0版本有点低,但是折腾不如先配上用了,以后有需要再说。 (后话:py 3.9 + pytorch 1.12.1 + cuda 11.3_cudnn 8_0 + pytorch3d 0.7.1 + CUB 1.11.0 也OK的) 1.1 创建新环境 1.2 找之前的pytorch版本,我这里直接给出我的选择的版本 一行行复制粘贴运行下面的命令 1.3 下载对应

    2024年02月08日
    浏览(26)
  • Pytorch3d中的倒角损失函数Chamfer Distance Loss的用法(pytorch3d.loss.chamfer_distance)

    API文档在这里 源码在这里 之前看到的一个干货满满的Pytorch3D安装指导与简单例子的帖子在这里  官方tutorials中提到的用法在下面这个代码块里面,前面后面东西都挺多的就把和chamfer_distance相关的摘到了这里

    2024年02月15日
    浏览(29)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包