CUDA11.3编译pytorch2.0.1报错:error: ‘nvmlProcessInfo_v1_t’ was not declared in this scope

发布时间 2023-08-09 13:12:38作者: Death_Knight

问题如题:

CUDA11.3编译pytorch2.0.1报错:error: ‘nvmlProcessInfo_v1_t’ was not declared in this scope

 

 

解决方法参考:

https://github.com/pytorch/pytorch/issues/100618

 

 

 

简单来说就是pytorch2.0.1源码不支持cuda11.3,需要改源码或者更新cuda版本,这里的建议是更新cuda版本,如使用pytorch官方建议的CUDA11.8版本。

 

 

 

 

==============================================