Jetson TX2安装PyTorch 1.11失败,自动安装1.8.0求助
解决Jetson TX2上PyTorch 1.11安装后自动变为1.8.0的问题
环境信息
- Nvidia Jetson TX2
- JetPack 4.6.5
- L4T Version 32.7.5
- Python 3.6.9
- Ubuntu 18.04
问题描述
在虚拟环境中尝试安装PyTorch 1.11版本,已执行pip3 uninstall torch和pip3 cache purge清理旧版本及缓存,通过指定whl包安装时日志显示处理1.11版本包,但最终提示成功安装PyTorch 1.8.0。
解决方案
1. 确认虚拟环境激活状态
确保安装操作在目标虚拟环境中进行,执行以下命令验证:
echo $VIRTUAL_ENV
若输出为空,说明未激活虚拟环境,以venv为例,重新激活:
source 你的虚拟环境路径/bin/activate
2. 校验下载的whl包完整性
检查本地whl包是否为正确的PyTorch 1.11版本,执行命令查看包详情:
pip3 show --file torch-1.11.0a0+17540c5+nv22.01-cp36-cp36m-linux_aarch64.whl
若输出中版本信息不匹配1.11相关内容,说明包下载损坏,重新执行wget命令下载,或手动确认包大小(该包约1.7GB)。
3. 强制安装并禁用缓存
使用--force-reinstall强制覆盖现有版本,--no-cache-dir避免调用任何缓存文件,执行命令:
pip3 install --force-reinstall --no-cache-dir numpy torch-1.11.0a0+17540c5+nv22.01-cp36-cp36m-linux_aarch64.whl
4. 排查依赖冲突
安装时仔细查看日志中的依赖提示,若存在torchvision、torchaudio等包依赖PyTorch 1.8.0,会触发自动降级。需先安装与PyTorch 1.11配套的对应版本依赖包,再重新安装PyTorch。
5. 验证最终安装版本
安装完成后,执行以下命令确认版本:
python3 -c "import torch; print(torch.__version__)"
内容的提问来源于stack exchange,提问作者Sajid Hussain
相关产品推荐
相关产品推荐

