1. PyTorch 安装
-
PyTorch 官网:PyTorch 官网
-
最新版本的 PyTorch:最新版本的 PyTorch
以 cuda 12.1,Python 3.10,Pytorch 2.1.2 为例
# conda 安装
conda install pytorch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 pytorch-cuda=12.1 -c pytorch -c nvidia
# pip 安装
pip install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu121
2. PyTorch3D 安装
- 由于直接使用 pip 进行PyTorch3D 安装会报错,所以选择在官网直接下载对应的安装包进行安装。
- 进入 Anaconda 官网找到 PyTorch3D:PyTorch3D 官网
找到和 cuda 以及 Python 版本对应的 Pytorch3D 版本,找到正确的版本后,右键绿色的文件名选择复制链接地址。暂时没有找到pytorch2.1.2版本,所以选择了pyt210版本。当然,如果没有找到对应你环境的版本,也可以重新创建一个对应的环境,再进行pytorch和pytorch3D的安装即可。
2.1 安装PyTorch3D
使用 conda install
+ 复制的 PyTorch3D 链接
如 安装对应 Python 3.10 cuda 12.1 Pytorch 2.1.2 版本的 PyTorch3D,使用如下命令:
conda install https://anaconda.org/pytorch3d/pytorch3d/0.7.5/download/linux-64/pytorch3d-0.7.5-py310_cu121_pyt210.tar.bz2
2.2 查看是否安装成功
conda list
成功安装!文章来源:https://www.toymoban.com/news/detail-844950.html
参考:
PyTorch3D 安装记录文章来源地址https://www.toymoban.com/news/detail-844950.html
到了这里,关于PyTorch3D 安装的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!