pip安装pytorch3d时如何快速确定版本

最新推荐文章于 2025-03-27 19:55:59 发布

文三路张同学

最新推荐文章于 2025-03-27 19:55:59 发布

阅读量367

点赞数 1

分类专栏：我的科研之路~ 文章标签： pip pytorch 人工智能

本文链接：https://blog.youkuaiyun.com/qq_36160277/article/details/140247509

版权

我的科研之路~ 专栏收录该内容

46 篇文章

订阅专栏

import sys
import torch
pyt_version_str=torch.__version__.split("+")[0].replace(".", "")
version_str="".join([
    f"py3{sys.version_info.minor}_cu",
    torch.version.cuda.replace(".",""),
    f"_pyt{pyt_version_str}"
])
print('pip install fvcore iopath')
print('pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/%s/download.html' % version_str)