Cuda版本不一致，编译apex报错

最新推荐文章于 2024-07-24 00:02:53 发布

番茄就要炒鸡蛋

最新推荐文章于 2024-07-24 00:02:53 发布

阅读量1.7k

点赞数

分类专栏：深度学习环境配置文章标签：计算机视觉深度学习

本文链接：https://blog.csdn.net/hejunran/article/details/125641624

版权

深度学习环境配置专栏收录该内容

5 篇文章 0 订阅

订阅专栏

解决办法

注释掉setup.py代码中的第108行，cuda版本检验，这样编译就可以了

if "--cuda_ext" in sys.argv:
    from torch.utils.cpp_extension import CUDAExtension
    sys.argv.remove("--cuda_ext")

    if torch.utils.cpp_extension.CUDA_HOME is None:
        raise RuntimeError("--cuda_ext was requested, but nvcc was not found.  Are you sure your environment has nvcc available?  If you're installing within a container from https://hub.docker.com/r/pytorch/pytorch, only images whose names contain 'devel' will provide nvcc.")
    else:
        # 注释掉这句
        # check_cuda_torch_binary_vs_bare_metal(torch.utils.cpp_extension.CUDA_HOME)