app/relay/interpreters.js


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

export default {
	pytorch: {
		cmd: process.env.PYTORCH_BIN,
		gpu: true,
	},
	tensorflow: {
		cmd: process.env.TENSORFLOW_BIN,
		gpu: true,
	},
	python: {
		cmd: process.env.PYTHON_BIN,
		gpu: false,
	},
	bash: {
		cmd: process.env.BASH_BIN || '/bin/bash',
		gpu: false,
	}
	perl: {
		cmd: process.env.PERL_BIN || '/usr/bin/perl',
		gpu: false,
	}
}