Skip to content

Commit

Permalink
FIX makefile
Browse files Browse the repository at this point in the history
  • Loading branch information
@matbun
matbun committed May 3, 2024
1 parent dc9b536 commit f7d30a8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions Makefile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,14 +15,14 @@ torch-env-cpu: env-files/torch/generic_torch.sh
tensorflow-env: env-files/tensorflow/generic_tf.sh
env ENV_NAME=.venv-tf \
bash -c 'bash env-files/tensorflow/generic_tf.sh'
.venv-tf/bin/horovodrun --check-build
@#.venv-tf/bin/horovodrun --check-build

# Install TensorFlow env (without GPU support: Horovod has not NCCL support)
tensorflow-env-cpu: env-files/tensorflow/generic_tf.sh
env ENV_NAME=.venv-tf \
NO_CUDA=1 \
bash -c 'bash env-files/tensorflow/generic_tf.sh'
.venv-tf/bin/horovodrun --check-build
@#.venv-tf/bin/horovodrun --check-build

test:
.venv-pytorch/bin/pytest -v tests/ -m "not slurm"
Expand Down

0 comments on commit f7d30a8

Please sign in to comment.