1
1
using PyCall
2
2
using Conda
3
3
4
- const cur_version = " 1.2 .0"
5
- const cur_py_version = " 1.2 .0"
4
+ const cur_version = " 1.4 .0"
5
+ const cur_py_version = " 1.4 .0"
6
6
7
7
8
8
65
65
66
66
base = dirname (@__FILE__ )
67
67
download_dir = joinpath (base, " downloads" )
68
+ lib_dir = joinpath (download_dir, " lib" )
68
69
bin_dir = joinpath (base, " usr/bin" )
69
70
70
71
if ! isdir (download_dir)
@@ -75,60 +76,12 @@ if !isdir(bin_dir)
75
76
run (` mkdir -p $bin_dir ` )
76
77
end
77
78
78
-
79
79
function download_and_unpack (url)
80
- tensorflow_zip_path = joinpath (base, " downloads/tensorflow.zip " )
80
+ tensorflow_zip_path = joinpath (base, " downloads/tensorflow.tar.gz " )
81
81
# Download
82
82
download (url, tensorflow_zip_path)
83
83
# Unpack
84
- try
85
- run (` unzip -o $(tensorflow_zip_path) ` )
86
- catch err
87
- if ! isfile (joinpath (base, " libtensorflow_c.so" ))
88
- throw (err)
89
- else
90
- warn (" Problem unzipping: $err " )
91
- end
92
- end
93
- end
94
-
95
-
96
- @static if is_apple ()
97
- download_and_unpack (" https://storage.googleapis.com/malmaud-stuff/tensorflow_mac_$cur_version .zip" )
98
- mv (" libtensorflow.so" , " usr/bin/libtensorflow.dylib" , remove_destination= true )
99
- end
100
-
101
- @static if is_linux ()
102
- if use_gpu
103
- url = " https://storage.googleapis.com/malmaud-stuff/tensorflow_linux_$cur_version .zip"
104
- else
105
- url = " https://storage.googleapis.com/malmaud-stuff/tensorflow_linux_cpu_$cur_version .zip"
106
- end
107
- download_and_unpack (url)
108
- mv (" libtensorflow.so" , " usr/bin/libtensorflow.so" , remove_destination= true )
109
- end
110
-
111
-
112
-
113
- # When TensorFlow issue #8669 is closed and a new version is released, use the official release binaries
114
- # of the TensorFlow C library.
115
- # see https://github.com/tensorflow/tensorflow/issues/8669
116
- # and then replacing the blocks above in this section below with:
117
- #=
118
- function download_and_unpack(url)
119
- tensorflow_zip_path = joinpath(base, "downloads/tensorflow.zip")
120
- # Download
121
- download(url, tensorflow_zip_path)
122
- # Unpack
123
- try
124
- run(`tar -xvzf $(tensorflow_zip_path) --strip-components=2 ./lib/libtensorflow.so`)
125
- catch err
126
- if !isfile(joinpath(base, "libtensorflow_c.so"))
127
- throw(err)
128
- else
129
- warn("Problem unzipping: $err")
130
- end
131
- end
84
+ run (` tar -xzf $tensorflow_zip_path -C downloads` )
132
85
end
133
86
134
87
@static if is_apple ()
138
91
url = " https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-darwin-x86_64-$cur_version .tar.gz"
139
92
end
140
93
download_and_unpack (url)
141
- mv("libtensorflow.so", "usr/bin/libtensorflow.dylib", remove_destination=true)
94
+ mv (" $lib_dir /libtensorflow.so" , " usr/bin/libtensorflow.dylib" , remove_destination= true )
95
+ mv (" $lib_dir /libtensorflow_framework.so" , " usr/bin/libtensorflow_framework.so" , remove_destination= true )
142
96
end
143
97
144
98
@static if is_linux ()
148
102
url = " https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-$cur_version .tar.gz"
149
103
end
150
104
download_and_unpack (url)
151
- mv("libtensorflow.so", "usr/bin/libtensorflow.so", remove_destination=true)
105
+ mv (" $lib_dir /libtensorflow.so" , " usr/bin/libtensorflow.so" , remove_destination= true )
106
+ mv (" $lib_dir /libtensorflow_framework.so" , " usr/bin/libtensorflow_framework.so" , remove_destination= true )
152
107
end
153
- =#
108
+
0 commit comments