Skip to content

GitLab

P
pytorch
Commit d9d6fd1b authored by Vladimir Bashkirtsev's avatar Vladimir Bashkirtsev
Browse files
Options

Enable CUDA for reComputer Industrial J4012

parent 8f7fe255
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
Makefile
View file @ d9d6fd1b
......@@ -10,7 +10,7 @@ j4012:
swapon swap
tar xf pytorch-v2.8.0.tar.gz
cd pytorch-v2.8.0 && USE_CUDA=1 USE_NCCL=0 USE_DISTRIBUTED=0 python3 setup.py install
cd pytorch-v2.8.0 && PATH=/build/usr/bin:$$PATH USE_CUDA=1 CUDA_HOME=/build/usr TORCH_CUDA_ARCH_LIST=8.7 USE_NCCL=0 USE_DISTRIBUTED=0 python3 setup.py install
rm -rf pytorch-v2.8.0
swapoff swap
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment