Build with new version of PyTorch fails

[davesean]

Due to the limitations of the PyTorch version of the prebuilt container, I wanted to build it myself by running:
docker build -t euclides_package package/

Even after modifying the whole Dockerfile, I can't get it to create the container successfully. Latest error was

CMake Error at CMakeLists.txt:31 (find_package):
By not providing "FindTorch.cmake" in CMAKE_MODULE_PATH this project has
asked CMake to find a package configuration file provided by "Torch", but
CMake did not find one.

Could not find a package configuration file provided by "Torch" with any of
the following names:

TorchConfig.cmake
torch-config.cmake

Add the installation prefix of "Torch" to CMAKE_PREFIX_PATH or set
"Torch_DIR" to a directory containing one of the above files. If "Torch"
provides a separate development package or SDK, be sure it has been
installed.

Even when adding these paths to the Dockerfile, it still fails to find them. Has anyone been able to build this container? Could you share your Dockerfile?