Try fix apex non-working on all presets (#127)

Author: Artem Yushkovskiy

Makefile

@@ -32,6 +32,11 @@ image_pip_list:
 e2e_neuro_push:
 	neuro push $(IMAGE_NAME):built $(TEST_IMAGE)
 
+
+.PHONY: test_e2e
+test_e2e: test_e2e_pytorch test_e2e_tensorflow test_e2e_apex_gpu_small test_e2e_apex_gpu_large test_e2e_dependencies
+	@echo "e2e tests succeeded!"
+
 TEST_PRESET=cpu-small
 TEST_CMD=
 .PHONY: _test_e2e
@@ -46,14 +51,25 @@ _test_e2e:
 
 .PHONY: test_e2e_pytorch
 test_e2e_pytorch: TEST_CMD=python /var/storage/gpu_pytorch.py
-test_e2e_pytorch: TEST_PRESET=gpu-small
+test_e2e_pytorch: TEST_PRESET=gpu-small-p
 test_e2e_pytorch: _test_e2e
 
 .PHONY: test_e2e_tensorflow
 test_e2e_tensorflow: TEST_CMD=python /var/storage/gpu_tensorflow.py
-test_e2e_tensorflow: TEST_PRESET=gpu-small
+test_e2e_tensorflow: TEST_PRESET=gpu-small-p
 test_e2e_tensorflow: _test_e2e
 
+.PHONY: test_e2e_apex_gpu_small
+test_e2e_apex_gpu_small: TEST_CMD=python /var/storage/gpu_apex.py
+test_e2e_apex_gpu_small: TEST_PRESET=gpu-small-p
+test_e2e_apex_gpu_small: _test_e2e
+test_e2e_tensorflow: _test_e2e
+
+.PHONY: test_e2e_apex_gpu_large
+test_e2e_apex_gpu_large: TEST_CMD=python /var/storage/gpu_apex.py
+test_e2e_apex_gpu_large: TEST_PRESET=gpu-large-p
+test_e2e_apex_gpu_large: _test_e2e
+
 .PHONY: test_e2e_dependencies
 test_e2e_dependencies: TEST_CMD=bash /var/storage/dependencies.sh
 test_e2e_dependencies: _test_e2e

targets/python37-jupyter-pytorch-tensorflow-jupyterlab/Dockerfile

@@ -171,6 +171,10 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \
 # Apex for PyTorch mixed precision training
 # ==================================================================
 
+# problem: https://github.com/neuro-inc/ml-kinect-pose-estimation/issues/59
+# solution: https://github.com/NVIDIA/apex/issues/842
+ENV TORCH_CUDA_ARCH_LIST 6.0;7.0
+
 # Somehow Apex does not use releases, and current master fails to build
 # (commit 1f2aa9156547377a023932a1512752c392d9bbdf on Apr 23, 2020).
 # So we fix installation to the version Apr 21, 2020