File tree Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Original file line number Diff line number Diff line change @@ -75,7 +75,7 @@ defmodule Torchx.MixProject do
75
75
76
76
defp libtorch_config ( ) do
77
77
target = System . get_env ( "LIBTORCH_TARGET" , "cpu" )
78
- version = System . get_env ( "LIBTORCH_VERSION" , "2.1 .0" )
78
+ version = System . get_env ( "LIBTORCH_VERSION" , "2.4 .0" )
79
79
env_dir = System . get_env ( "LIBTORCH_DIR" )
80
80
81
81
% {
@@ -135,18 +135,15 @@ defmodule Torchx.MixProject do
135
135
"https://download.pytorch.org/libtorch/#{ libtorch_config . target } /libtorch-cxx11-abi-shared-with-deps-#{ libtorch_config . version } %2B#{ libtorch_config . target } .zip"
136
136
137
137
{ :unix , :darwin } ->
138
- # MacOS
139
- # pytorch only provides official pre-built binaries for x86_64
140
138
case List . to_string ( :erlang . system_info ( :system_architecture ) ) do
141
139
"x86_64" <> _ ->
142
140
"https://download.pytorch.org/libtorch/#{ libtorch_config . target } /libtorch-macos-#{ libtorch_config . version } .zip"
143
141
144
142
_ ->
145
- "https://github.com/mlverse/ libtorch-mac-m1/releases/download/LibTorch/ libtorch-v #{ libtorch_config . version } .zip"
143
+ "https://download.pytorch.org/ libtorch/ #{ libtorch_config . target } / libtorch-macos-arm64- #{ libtorch_config . version } .zip"
146
144
end
147
145
148
146
{ :win32 , :nt } ->
149
- # Windows
150
147
"https://download.pytorch.org/libtorch/#{ libtorch_config . target } /libtorch-win-shared-with-deps-#{ libtorch_config . version } %2B#{ libtorch_config . target } .zip"
151
148
152
149
os ->
You can’t perform that action at this time.
0 commit comments