ubuntu24下安装pytorch3d

pytoch3d的安装,首先根据自己torch和cuda的版本信息去下面的网站下载对应的pytroch3d。

https://anaconda.org/pytorch3d/pytorch3d/files?page=2

在这里插入图片描述
下载到本地之后使用下面的命令进行安装即可

conda install pytorch3d-0.7.5-py39_cu121_pyt210.tar.bz2

一般情况下,安装完成之后会缺少两个包,安装上即可。

pip install fvcore
pip install iopath

最后,在终端通过下面的命令进行简单测试。

python -c "import torch; import pytorch3d; print('PyTorch version:', torch.__version__); print('PyTorch3D version:', pytorch3d.__version__)"

在这里插入图片描述
出现上述版本信息,表明安装成功。

要在 Ubuntu 20.04 上搭建 PyTorch 3D 模型训练环境,你可以按照以下步骤进行操作: 1. 安装 CUDA 和 cuDNN(如果你的 GPU 支持 CUDA): - 首先,检查你的 GPU 支持的 CUDA 版本,并从 NVIDIA 官方网站下载对应版本的 CUDA Toolkit。 - 安装 CUDA Toolkit,按照官方文档提供的说明进行安装。 - 在安装 CUDA Toolkit 后,下载对应版本的 cuDNN,并按照官方文档提供的说明进行安装。 2. 安装 Anaconda(可选): - 访问 Anaconda 官方网站并下载适用于 Linux 的 Anaconda 安装包。 - 打开终端,导航到下载的 Anaconda 安装包所在的目录,并执行以下命令安装 Anaconda: ``` bash Anaconda3-2021.05-Linux-x86_64.sh ``` - 按照安装程序的提示进行安装。 3. 创建并激活虚拟环境: - 打开终端,执行以下命令创建一个名为 "pytorch" 的虚拟环境: ``` conda create -n pytorch python=3.8 ``` - 激活虚拟环境: ``` conda activate pytorch ``` 4. 安装 PyTorch 和 torchvision: - 在激活的虚拟环境中,执行以下命令安装 PyTorch 和 torchvision: ``` conda install pytorch torchvision torchaudio cudatoolkit=xx.x -c pytorch ``` (请使用你安装的 CUDA 版本的正确替代项 xx.x) 5. 安装其他依赖项: - 执行以下命令安装其他常用的 PyTorch 相关依赖项: ``` conda install matplotlib numpy pandas ``` 这样,你就可以在 Ubuntu 20.04 上成功搭建 PyTorch 3D 模型训练环境了。记得在使用 PyTorch 时,始终在激活的虚拟环境中运行代码。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

是小峰呀

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值