Linux服务器安装pytorch3d完整流程

接上一篇安装tiny-cuda-nn完整流程,本人现在记录一下安装pytorch3d完整流程,同时供各位一块安装,省去一些不必要的时间。

首先需要提醒的是,虽然其他博客提到直接使用conda install和pip install十分的方便,但是由于这两种安装的包的编译环境大概率和你自己的虚拟环境不一样,会导致安装的包无法使用,所以强烈建议从git上自己手动进行安装!!

1、命令行安装(最方便快捷)

直接从github上面自动进行pip安装:

pip install git+https://github.com/facebookresearch/pytorch3d.git

同时再次提醒各位,如果发生服务器连接不上github的情况,建议直接换成gitee进行安装,此处放出我自己clone的仓库链接:

https://gitee.com/richard_chenRuiyang/pytorch3d

同时本人想提醒各位,如果从github上面clone包失败,可以先将包迁移到自己的gitee上,亲测十分有效,可以减少很多时间,具体可以参考如下文章:

【经验】Git|Linux终端git太慢,改hosts、复制文件夹、用镜像源?不不不不不

2、手动安装(稍微麻烦)

为什么我的两篇文章中都会提到使用手动安装,是因为在我的服务器上,build wheel总是失败,总是需要一个pyproject,所以我的安装都是使用手动的方法的。

其次还需要说明,根据其他博客,pytorch3d源码是安装不成功的,需要使用修改后的代码,经实测修改后的包没有问题,https://gitee.com/richard_chenRuiyang/pytorch3d链接中的包是修改后的包,可以直接使用。

(1)将pytorch3d包git clone到虚拟环境中:

git clone https://gitee.com/richard_chenRuiyang/pytorch3d

(2)安装pytorch3d的依赖包:

conda install fvcore  iopath

conda install -c bottler nvidiacub

(3)确认系统的gcc和g++版本

安装pytorch3d包的时候gcc和g++版本需要是7.x,如果不符合要求在网上很容易搜到gcc,g++版本切换的方法,在此不过多赘述。

(4)修改setup.py文件

在我安装包的时候,总是出现一些关于ninja的问题,在参考了这篇文章之后找到了解决方法:​​​​​​出现错误“subprocess.CalledProcessError: Command ‘[‘ninja‘, ‘-v‘]‘ returned non-zero exit status 1”解决方法

所以在此步,我们需要打开pytorch3d/setup.py文件,定位到最后一行,将:

cmdclass={'build_ext': BuildExtension}

改为:

cmdclass={'build_ext': BuildExtension.with_options(use_ninja=False)}

这样在之后的安装过程中就不会出现问题了。

(5)安装pytorch3d包

使用以下指令即可正确安装pytorch3d包:

python setup.py install

(6)检查是否安装成功

在安装结束后可以通过pip list等指令检查是否成功安装包。

在我的安装过程中,遇到了很多奇奇怪怪的问题,花了两天时间最终成功装好,在此感谢csdn各位大佬的博客,也希望各位能够通过我的博客很快安装好pytorch3d包。在此引用其他博客的一句话,在安装pytorch3d的时候不要一直使用一个方法,一个方法进行不下去的时候迅速更换其他方法进行安装。

在此非常感谢csdn的前辈,本篇博客参考以下文章:

ubuntu安装pytorch3d

​​​​​​出现错误“subprocess.CalledProcessError: Command ‘[‘ninja‘, ‘-v‘]‘ returned non-zero exit status 1”解决方法

  • 21
    点赞
  • 23
    收藏
    觉得还不错? 一键收藏
  • 6
    评论
### 回答1: 要在 Linux 服务器安装 PyTorch,您可以按照以下步骤操作: 1. 确保您的服务器已经安装Python 3,并且已经安装了包管理工具 pip。 2. 打开终端,使用 pip 安装 PyTorch。可以通过以下命令来安装: ``` pip install torch ``` 3. 安装 PyTorch 的同时也会安装其依赖项,包括 NumPy 和 CUDA 等。如果您需要特定版本的 PyTorch 或者需要使用 GPU 运行 PyTorch,可以在安装时指定版本或者使用额外的选项。 安装完成后,您就可以在 Linux 服务器上使用 PyTorch 进行深度学习和机器学习等任务了。 ### 回答2: 为了在Linux服务器安装PyTorch,需要完成以下几个步骤: 1. 安装conda或者miniconda:首先需要安装conda或者miniconda,这是因为PyTorch可以通过conda进行安装。在安装过程中,conda安装Python和相关的库。 2. 创建conda环境:在安装PyTorch之前,应该创建一个conda环境,以避免系统中存在版本冲突。可以使用以下命令创建环境: ``` conda create --name pytorch_env python=3.7 ``` 3. 激活conda环境:创建环境之后,需要激活它。可以使用以下命令激活环境: ``` conda activate pytorch_env ``` 4. 安装PyTorch:然后需要安装PyTorch,可以使用以下命令安装: ``` conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch ``` 由于PyTorch支持CPU和GPU版本,以上命令中的"cudatoolkit"参数可以根据需要指定。例如,可以将其设置为"cpu"以安装CPU版本的PyTorch。 5. 验证PyTorch安装安装完成后,需要验证PyTorch是否正确安装。可以使用以下简单的Python代码进行验证: ``` import torch print(torch.__version__) ``` 如果输出了正确的PyTorch版本号,则说明安装成功。 在安装PyTorch时,还应该注意一些细节。例如,应该确保conda环境的Python版本与需要的PyTorch版本兼容,否则可能会出现不兼容问题。此外,应该根据需要选择正确的PyTorch版本,以支持所需的功能或性能。最后,还应该确保Linux服务器上的GPU驱动程序正确安装,以便使用GPU加速的PyTorch版本。 ### 回答3: 安装PyTorchLinux服务器上是一项不容小视的任务,因为PyTorch是基于Python的强大深度学习框架,可以使你的机器学习过程更加快速和高效。在开始安装PyTorch之前,需要谨慎分析您的服务器配置和Python环境,以确保您的安装是顺利的。 步骤一:安装Python环境 首先,您应该在Linux服务器安装Python环境。通过如下命令安装: sudo apt-get update sudo apt-get install python3 sudo apt-get install python3-pip 步骤二:安装PyTorch 有几种安装PyTorch的方式,可以从Anacondapip或源代码安装。下面我们以pip方式为例。 使用以下命令安装PyTorchpip3 install torch torchvision 请注意,这个过程可能需要一些时间,具体时间取决于服务器的性能和网络连接速度。 步骤三:测试PyTorch是否安装成功 完成安装后,您可以使用以下命令测试PyTorch是否安装成功: python3 import torch print(torch.__version__) 如果输出的版本号与您安装的版本号相同,则表明PyTorch安装成功。 总结 安装PyTorchLinux服务器上可以使您的深度学习任务更高效,但是需要谨慎检查您的服务器配置和Python环境。要安装PyTorch,首先要安装Python环境,然后使用pipLinux服务器安装PyTorch,最后测试是否安装成功。如果您遇到任何问题,请仔细检查错误信息并进行相应的处理。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值