Skip to content

Commit a4e9c9d

Browse files
committedJan 14, 2025
modify so the system settings do not interact with this
1 parent 0389855 commit a4e9c9d

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed
 

‎src/main/java/io/bioimage/modelrunner/pytorch/JavaWorker.java

+1
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ public class JavaWorker {
3030
*/
3131
public static void main(String[] args) {
3232
System.clearProperty("PYTORCH_FLAVOR");
33+
System.clearProperty("java.library.path");
3334

3435
try(Scanner scanner = new Scanner(System.in)){
3536
PytorchInterface pi;

‎src/main/java/io/bioimage/modelrunner/pytorch/PytorchInterface.java

+8-1
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@
5959
import java.util.LinkedList;
6060
import java.util.List;
6161
import java.util.Map;
62+
import java.util.Properties;
6263

6364
import com.google.gson.Gson;
6465

@@ -145,7 +146,13 @@ private Service getRunner() throws IOException, URISyntaxException {
145146
String[] argArr = new String[args.size()];
146147
args.toArray(argArr);
147148

148-
return new Service(new File("."), argArr);
149+
Service service = new Service(new File("."), argArr);
150+
service.setEnvVar("CUDA_HOME", null);
151+
service.setEnvVar("CUDA_PATH", null);
152+
service.setEnvVar("LD_LIBRARY_PATH", null);
153+
service.setEnvVar("DYLD_LIBRARY_PATH", null);
154+
service.setEnvVar("PATH", null);
155+
return service;
149156
}
150157

151158
/**

0 commit comments

Comments
 (0)