cmake指定cuda编译

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

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

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

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