关于pytorch1.9.0中cpp_extension的默认CUDA版本是11.0的问题
·
今天在用CUDAExtension编译一个算子的时候，发现它默认的CUDA路径指向的是11.0版本，而我的CUDA版本是11.1.
这会导致使用nvcc编译时，会报错No such file or directory错误。
这时我们只需要将CUDA的路径指向我们安装的路径即可，比如下面这样
torch.utils.cpp_extension.CUDA_HOME = "/usr/local/cuda-11.1"
就能够将CUDA路径纠正。
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
0
0- 0
已为社区贡献2条内容
所有评论(0)