When I try to compile Pytorch with Caffe2 on Ubuntu, I get one problem like this:
[ 98%] Building CXX object test_tensorexpr/CMakeFiles/test_tensorexpr.dir/test_ir_printer.cpp.o
/home/xiangzhaorui/miniconda3/envs/swts/bin/…/lib/gcc/x86_64-conda-linux-gnu/10.3.0/…/…/…/…/x86_64-conda-linux-gnu/bin/ld: CMakeFiles/context_gpu_test.dir/core/context_gpu_test.cc.o: in function caffe2::getStreamForHandle(cublasContext*)': context_gpu_test.cc:(.text._ZN6caffe218getStreamForHandleEP13cublasContext+0x2c): undefined reference to
cublasGetStream_v2’
/home/xiangzhaorui/miniconda3/envs/swts/bin/…/lib/gcc/x86_64-conda-linux-gnu/10.3.0/…/…/…/…/x86_64-conda-linux-gnu/bin/ld: CMakeFiles/context_gpu_test.dir/core/context_gpu_test.cc.o: in function caffe2::CUDAContext::curand_generator()': context_gpu_test.cc:(.text._ZN6caffe211CUDAContext16curand_generatorEv[_ZN6caffe211CUDAContext16curand_generatorEv]+0x56): undefined reference to
curandSetStream’
/home/xiangzhaorui/miniconda3/envs/swts/bin/…/lib/gcc/x86_64-conda-linux-gnu/10.3.0/…/…/…/…/x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe211CUDAContext16curand_generatorEv[_ZN6caffe211CUDAContext16curand_generatorEv]+0xcc): undefined reference to curandCreateGenerator' /home/xiangzhaorui/miniconda3/envs/swts/bin/../lib/gcc/x86_64-conda-linux-gnu/10.3.0/../../../../x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe211CUDAContext16curand_generatorEv[_ZN6caffe211CUDAContext16curand_generatorEv]+0xec): undefined reference to
curandSetPseudoRandomGeneratorSeed’
/home/xiangzhaorui/miniconda3/envs/swts/bin/…/lib/gcc/x86_64-conda-linux-gnu/10.3.0/…/…/…/…/x86_64-conda-linux-gnu/bin/ld: CMakeFiles/context_gpu_test.dir/core/context_gpu_test.cc.o: in function caffe2::CUDAContextTest_TestSameThreadSameObject_Test::TestBody()': context_gpu_test.cc:(.text._ZN6caffe245CUDAContextTest_TestSameThreadSameObject_Test8TestBodyEv+0x685): undefined reference to
cublasCreate_v2’
/home/xiangzhaorui/miniconda3/envs/swts/bin/…/lib/gcc/x86_64-conda-linux-gnu/10.3.0/…/…/…/…/x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe245CUDAContextTest_TestSameThreadSameObject_Test8TestBodyEv+0x6a8): undefined reference to cublasSetPointerMode_v2' /home/xiangzhaorui/miniconda3/envs/swts/bin/../lib/gcc/x86_64-conda-linux-gnu/10.3.0/../../../../x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe245CUDAContextTest_TestSameThreadSameObject_Test8TestBodyEv+0x6d9): undefined reference to
cublasSetStream_v2’
/home/xiangzhaorui/miniconda3/envs/swts/bin/…/lib/gcc/x86_64-conda-linux-gnu/10.3.0/…/…/…/…/x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe245CUDAContextTest_TestSameThreadSameObject_Test8TestBodyEv+0x7d5): undefined reference to cublasCreate_v2' /home/xiangzhaorui/miniconda3/envs/swts/bin/../lib/gcc/x86_64-conda-linux-gnu/10.3.0/../../../../x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe245CUDAContextTest_TestSameThreadSameObject_Test8TestBodyEv+0x7f8): undefined reference to
cublasSetPointerMode_v2’
/home/xiangzhaorui/miniconda3/envs/swts/bin/…/lib/gcc/x86_64-conda-linux-gnu/10.3.0/…/…/…/…/x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe245CUDAContextTest_TestSameThreadSameObject_Test8TestBodyEv+0x829): undefined reference to cublasSetStream_v2' /home/xiangzhaorui/miniconda3/envs/swts/bin/../lib/gcc/x86_64-conda-linux-gnu/10.3.0/../../../../x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe245CUDAContextTest_TestSameThreadSameObject_Test8TestBodyEv+0x935): undefined reference to
cublasCreate_v2’
/home/xiangzhaorui/miniconda3/envs/swts/bin/…/lib/gcc/x86_64-conda-linux-gnu/10.3.0/…/…/…/…/x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe245CUDAContextTest_TestSameThreadSameObject_Test8TestBodyEv+0x958): undefined reference to cublasSetPointerMode_v2' /home/xiangzhaorui/miniconda3/envs/swts/bin/../lib/gcc/x86_64-conda-linux-gnu/10.3.0/../../../../x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe245CUDAContextTest_TestSameThreadSameObject_Test8TestBodyEv+0x985): undefined reference to
cublasSetStream_v2’
/home/xiangzhaorui/miniconda3/envs/swts/bin/…/lib/gcc/x86_64-conda-linux-gnu/10.3.0/…/…/…/…/x86_64-conda-linux-gnu/bin/ld: CMakeFiles/context_gpu_test.dir/core/context_gpu_test.cc.o: in function caffe2::CUDAContextTest_TestSameThreadDifferntObjectIfDifferentDevices_Test::TestBody()': context_gpu_test.cc:(.text._ZN6caffe267CUDAContextTest_TestSameThreadDifferntObjectIfDifferentDevices_Test8TestBodyEv+0x4d5): undefined reference to
cublasCreate_v2’
/home/xiangzhaorui/miniconda3/envs/swts/bin/…/lib/gcc/x86_64-conda-linux-gnu/10.3.0/…/…/…/…/x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe267CUDAContextTest_TestSameThreadDifferntObjectIfDifferentDevices_Test8TestBodyEv+0x4f9): undefined reference to cublasSetPointerMode_v2' /home/xiangzhaorui/miniconda3/envs/swts/bin/../lib/gcc/x86_64-conda-linux-gnu/10.3.0/../../../../x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe267CUDAContextTest_TestSameThreadDifferntObjectIfDifferentDevices_Test8TestBodyEv+0x527): undefined reference to
cublasSetStream_v2’
/home/xiangzhaorui/miniconda3/envs/swts/bin/…/lib/gcc/x86_64-conda-linux-gnu/10.3.0/…/…/…/…/x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe267CUDAContextTest_TestSameThreadDifferntObjectIfDifferentDevices_Test8TestBodyEv+0x695): undefined reference to cublasCreate_v2' /home/xiangzhaorui/miniconda3/envs/swts/bin/../lib/gcc/x86_64-conda-linux-gnu/10.3.0/../../../../x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe267CUDAContextTest_TestSameThreadDifferntObjectIfDifferentDevices_Test8TestBodyEv+0x6b9): undefined reference to
cublasSetPointerMode_v2’
/home/xiangzhaorui/miniconda3/envs/swts/bin/…/lib/gcc/x86_64-conda-linux-gnu/10.3.0/…/…/…/…/x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe267CUDAContextTest_TestSameThreadDifferntObjectIfDifferentDevices_Test8TestBodyEv+0x6eb): undefined reference to cublasSetStream_v2' /home/xiangzhaorui/miniconda3/envs/swts/bin/../lib/gcc/x86_64-conda-linux-gnu/10.3.0/../../../../x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe267CUDAContextTest_TestSameThreadDifferntObjectIfDifferentDevices_Test8TestBodyEv+0x91d): undefined reference to
cublasCreate_v2’
/home/xiangzhaorui/miniconda3/envs/swts/bin/…/lib/gcc/x86_64-conda-linux-gnu/10.3.0/…/…/…/…/x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe267CUDAContextTest_TestSameThreadDifferntObjectIfDifferentDevices_Test8TestBodyEv+0x940): undefined reference to cublasSetPointerMode_v2' /home/xiangzhaorui/miniconda3/envs/swts/bin/../lib/gcc/x86_64-conda-linux-gnu/10.3.0/../../../../x86_64-conda-linux-gnu/bin/ld: context_gpu_test.cc:(.text._ZN6caffe267CUDAContextTest_TestSameThreadDifferntObjectIfDifferentDevices_Test8TestBodyEv+0x96d): undefined reference to
cublasSetStream_v2’
And this is the only problem that I’ve not solved. Does anyone know how to solve this?
ps:I compiled with Ubuntu 18.04 anaconda cuda 11.7 gcc 10.3 g++10.3