<html><head><meta http-equiv="Content-Type" content="text/html; charset=utf-8"></head><body style="word-wrap: break-word; -webkit-nbsp-mode: space; line-break: after-white-space;" class="">Hi everyone,<div class=""><br class=""></div><div class="">I had a bit of an adventure trying to build JAX on Auton and thought I’d document the right way to do it so that if others want they can as well. Here are the steps to do it on a machine that has CUDA 11 and cuDNN 8:</div><div class=""><br class=""></div><div class="">1. Go somewhere in your scratch</div><div class="">2. <span class="n" style="color: rgb(64, 64, 64); font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box;">git</span><span style="color: rgb(64, 64, 64); font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2;" class=""> </span><span class="n" style="color: rgb(64, 64, 64); font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box;">clone</span><span style="color: rgb(64, 64, 64); font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2;" class=""> </span><span class="n" style="color: rgb(64, 64, 64); font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box;">https</span><span class="p" style="color: rgb(64, 64, 64); font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box;">:</span><span class="o" style="font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box; color: rgb(102, 102, 102);">//</span><span class="n" style="color: rgb(64, 64, 64); font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box;">github</span><span class="o" style="font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box; color: rgb(102, 102, 102);">.</span><span class="n" style="color: rgb(64, 64, 64); font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box;">com</span><span class="o" style="font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box; color: rgb(102, 102, 102);">/</span><span class="n" style="color: rgb(64, 64, 64); font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box;">google</span><span class="o" style="font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box; color: rgb(102, 102, 102);">/</span><span class="n" style="color: rgb(64, 64, 64); font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box;">jax</span></div><div class="">3. Make a python environment with a python >= 3.6 using Conda or virtualenv</div><div class="">4. Install numpy, spicy, cython, six</div><div class="">5. Source /opt/rh/devtoolset-8/enable</div><div class="">6. Run <span class="n" style="color: rgb(64, 64, 64); font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box;">python</span><span style="color: rgb(64, 64, 64); font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2;" class=""> </span><span class="n" style="color: rgb(64, 64, 64); font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box;">build</span><span class="o" style="font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box; color: rgb(102, 102, 102);">/</span><span class="n" style="color: rgb(64, 64, 64); font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box;">build</span><span class="o" style="font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box; color: rgb(102, 102, 102);">.</span><span class="n" style="color: rgb(64, 64, 64); font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box;">py</span><span style="color: rgb(64, 64, 64); font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2;" class=""> </span><span class="o" style="font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box; color: rgb(102, 102, 102);">—</span><span class="n" style="color: rgb(64, 64, 64); font-family: SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", Courier, monospace; orphans: 2; widows: 2; box-sizing: border-box;">enable_cuda —cuda_path /usr/local/cuda —cudnn_path /usr/</span></div><div style="orphans: 2; widows: 2;" class=""><font color="#000000" face="Arial" class=""><span style="caret-color: rgb(64, 64, 64);" class="">7. Pip install -e build</span></font></div><div style="orphans: 2; widows: 2;" class=""><font color="#000000" face="Arial" class=""><span style="caret-color: rgb(64, 64, 64);" class="">8. Pip install -e .</span></font></div><div style="orphans: 2; widows: 2;" class=""><font color="#000000" face="Arial" class=""><span style="caret-color: rgb(64, 64, 64);" class=""><br class=""></span></font></div><div style="orphans: 2; widows: 2;" class=""><font color="#000000" face="Arial" class=""><span style="caret-color: rgb(64, 64, 64);" class="">I know this isn’t that complicated, but I figured it would save some effort if anyone else would like to use it.</span></font></div><div style="orphans: 2; widows: 2;" class=""><font color="#000000" face="Arial" class=""><span style="caret-color: rgb(64, 64, 64);" class=""><br class=""></span></font></div><div style="orphans: 2; widows: 2;" class=""><font color="#000000" face="Arial" class=""><span style="caret-color: rgb(64, 64, 64);" class="">Cheers,</span></font></div><div style="orphans: 2; widows: 2;" class=""><font color="#000000" face="Arial" class=""><span style="caret-color: rgb(64, 64, 64);" class="">Viraj</span></font></div></body></html>