Skip to content

Commit

Permalink
run on gpu
Browse files Browse the repository at this point in the history
  • Loading branch information
hlm628 committed Mar 23, 2024
1 parent 27b37bf commit eb62433
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions Makefile
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
GPU=0
CUDNN=0
CUDNN_HALF=0
OPENCV=0
GPU=1
CUDNN=1
CUDNN_HALF=1
OPENCV=1
AVX=0
OPENMP=0
LIBSO=0
Expand Down Expand Up @@ -57,7 +57,7 @@ OS := $(shell uname)
# ARCH= -gencode arch=compute_75,code=[sm_75,compute_75]

# Tesla A100 (GA100), DGX-A100, A30, A100, RTX 3080
# ARCH= -gencode arch=compute_80,code=[sm_80,compute_80]
ARCH= -gencode arch=compute_80,code=[sm_80,compute_80]

# GeForce RTX 30x0, 30x0 Ti, Tesla GA10x, RTX Axxxx, A2, A10, A16, A40
# ARCH= -gencode arch=compute_86,code=[sm_86,compute_86]
Expand Down

0 comments on commit eb62433

Please sign in to comment.