因为linux gpu机器上同时安装了cuda10.1和cuda11。并且环境变量设置的是cuda11,于是不指定特定版本的cuda时,使用的是cuda11,此时我刚好要编译依赖cuda10.1的库,同时又懒得去修改环境变量。

所以就在cmake中增加查找库路径了。

set(CUDA_TOOLKIT_ROOT_DIR /usr/local/cuda-10.1)

这样编译的程序就不会依赖cuda11,而是依赖cuda10.1了。

Logo

更多推荐