alplapose(pytorch version)安装配置

本文档详细介绍了在Ubuntu16.04上安装配置Alphapose(PyTorch版本)的步骤,包括解决ntpath依赖问题、安装PyTorch、下载模型权重、安装NVIDIA驱动、处理CUDA内存不足的错误,以及处理系统重启后黑屏的问题。通过这个过程,读者可以成功运行Alphapose并进行人体关键点检测。

Alphapose(pytorch version)github页面:https://github.com/MVIG-SJTU/AlphaPose/tree/pytorch

搞了双系统win10+ubuntu16.04
在ubuntu16.04内部运行alphapose代码

  • python2.7和python3.5,先安装pip3(python3-pip)
  • 下代码
git clone -b pytorch https://github.com/MVIG-SJTU/AlphaPose.git
  • 安装代码依赖项
pip3 install -r requirements.txt

注意:ntpath总是提示Could not find a version that satisfies the requirement ntpath
单独安装ntpath也没用,上网搜ntpath资料也是少之又少,最后在alphapose的issue中看到有人说ntpath not necessary。fine,不管它!

在 PyCharm 中配置 PyTorch 环境可以分为以下几个步骤,包括创建虚拟环境、安装 PyTorch 依赖包以及测试环境是否配置成功。以下是详细的操作说明: ### 创建虚拟环境 在 PyCharm 中创建一个新的项目,并选择一个合适的解释器。建议使用虚拟环境以避免不同项目的依赖冲突。在创建项目时,可以选择新建一个虚拟环境。 ### 安装 PyTorch 可以通过 PyCharm 的包管理功能来安装 PyTorch。打开 PyCharm,进入 `File` -> `Settings`(或者在项目中点击右下角的 `Python Interpreter`),然后点击 `+` 号来添加新的包。搜索 `torch` 和 `torchvision`,选择合适的版本进行安装。对于 Windows 用户,如果需要特定版本,可以从 PyTorch 官网获取安装命令,例如使用 pip 安装命令为 `pip install torch==1.4.0+cu101 torchvision==0.5.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html` [^2]。 ### 配置环境变量 通常情况下,安装 PyTorch 不需要额外配置环境变量,因为通过 pip 安装的包会自动处理好路径问题。但如果需要使用 CUDA 支持,则需要确保 NVIDIA 的驱动程序已经正确安装,并且 CUDA 工具包已经被添加到系统路径中。 ### 测试安装 安装完成后,可以通过运行一段简单的代码来验证 PyTorch 是否安装成功,以及是否能够检测到 GPU。例如,创建一个 Python 文件,输入以下代码: ```python import torch print("hello torch{}".format(torch.__version__)) flag = torch.cuda.is_available() print(flag) ngpu = 1 device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu") print(device) print(torch.cuda.get_device_name(0)) print(torch.rand(3, 3).cuda()) ``` 这段代码会输出 PyTorch 的版本号、是否可以使用 CUDA、当前使用的设备名称、GPU 名称以及在 GPU 上生成的一个随机矩阵 [^3]。 通过以上步骤,应该能够在 PyCharm 中成功安装配置 PyTorch 环境。
评论 8
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值