Skip to content

Commit 8e40eaf

Browse files
committed
Change PT_BUILD to cu111 in get_deps for GPU
1 parent a11bb24 commit 8e40eaf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

get_deps.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -192,7 +192,7 @@ PT_ARCH=x86_64
192192
if [[ $OS == linux ]]; then
193193
PT_OS=linux
194194
if [[ $GPU == 1 ]]; then
195-
PT_BUILD=cu110
195+
PT_BUILD=cu111
196196
fi
197197

198198
if [[ $ARCH == x64 ]]; then

0 commit comments

Comments
 (0)