Linear algebra methods rely on magma (e.g. here 19) so you won’t be able to use them. However, a lot of models don’t use these methods, so as long as you don’t need to e.g. calculate the log determinant of your weight parameter, you should be fine.


I would recommend to stick to pytorch/builder 65 for the magma build.
For CUDA10.2 you shouldn’t need any changes in the build script besides the CUDA version change, but let me know, if you get stuck. Also, you could have a look at the NGC PyTorch container 94, which ships with CUDA10.2.