在Jetson上跑一个PyTorch推理程序,安装完基础环境之后,根据CUDA版本安装PyTorch。
CUDA版本为11.4:
然后通过PyTorch官网获取安装命令行:
安装成功后,运行推理程序时出现题目中的Error:
测试CUDA,发现不可用:
最开始怀疑PyTorch的CUDA版本问题,于是升级到11.6,该问题依然存在。
后来的定位经过:
(1) JetPack与PyTorch的兼容性问题,参考链接
在Jetson上跑一个PyTorch推理程序,安装完基础环境之后,根据CUDA版本安装PyTorch。
CUDA版本为11.4:
然后通过PyTorch官网获取安装命令行:
安装成功后,运行推理程序时出现题目中的Error:
测试CUDA,发现不可用:
最开始怀疑PyTorch的CUDA版本问题,于是升级到11.6,该问题依然存在。
后来的定位经过:
(1) JetPack与PyTorch的兼容性问题,参考链接