Update .bazelrc
To make MediaPipe get TensorFlow’s CUDA settings copied the content from tensorflow's bazel.rc file into mediapipe's bazel.rc file
This commit is contained in:
		
							parent
							
								
									137e1cc763
								
							
						
					
					
						commit
						2eab5020a2
					
				
							
								
								
									
										10
									
								
								.bazelrc
									
									
									
									
									
								
							
							
						
						
									
										10
									
								
								.bazelrc
									
									
									
									
									
								
							| 
						 | 
					@ -16,6 +16,16 @@ build:linux --cxxopt=-std=c++17
 | 
				
			||||||
build:linux --host_cxxopt=-std=c++17
 | 
					build:linux --host_cxxopt=-std=c++17
 | 
				
			||||||
build:linux --copt=-w
 | 
					build:linux --copt=-w
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# This config refers to building with CUDA available. It does not necessarily
 | 
				
			||||||
 | 
					# mean that we build CUDA op kernels.
 | 
				
			||||||
 | 
					build:using_cuda --define=using_cuda=true
 | 
				
			||||||
 | 
					build:using_cuda --action_env TF_NEED_CUDA=1
 | 
				
			||||||
 | 
					build:using_cuda --crosstool_top=@local_config_cuda//crosstool:toolchain
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# This config refers to building CUDA op kernels with nvcc.
 | 
				
			||||||
 | 
					build:cuda --config=using_cuda
 | 
				
			||||||
 | 
					build:cuda --define=using_cuda_nvcc=true
 | 
				
			||||||
 | 
					
 | 
				
			||||||
# windows
 | 
					# windows
 | 
				
			||||||
build:windows --cxxopt=/std:c++17
 | 
					build:windows --cxxopt=/std:c++17
 | 
				
			||||||
build:windows --host_cxxopt=/std:c++17
 | 
					build:windows --host_cxxopt=/std:c++17
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
		Loading…
	
		Reference in New Issue
	
	Block a user