rpms/python-torch
3
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add cuda arches to build for

Browse source 
Suggested-by: Steve Royer <sroyer@redhat.com>
Signed-off-by: Tom Rix <trix@redhat.com>
This commit is contained in:
Tom Rix 2024-05-16 14:07:10 -06:00
commit 64cda83196
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
python-torch.spec
View file

@ -681,6 +681,8 @@ export BUILD_CAFFE2=ON
export CUDACXX=/usr/local/cuda/bin/nvcc
export CPLUS_INCLUDE_PATH=/usr/local/cuda/include
export USE_CUDA=ON
# The arches to build for
export TORCH_CUDA_ARCH_LIST="8.0 8.6 8.9 9.0"
%endif
%endif