Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/torch-sparce.git" did not exist on "9c3519b47f84725a0ffdf6b74601e1695139c089"
Commit 27b934a8 authored by rusty1s's avatar rusty1s
Browse files

test cuda

parent 765003dd
Hide whitespace changes
Inline Side-by-side
Showing with 10 additions and 6 deletions
.travis.yml
View file @ 27b934a8
language: shell language: shell
os: os:
- linux # - linux
# - osx # - osx
- windows - windows
...@@ -9,7 +9,7 @@ env: ...@@ -9,7 +9,7 @@ env:
global: global:
- CUDA_HOME=/usr/local/cuda - CUDA_HOME=/usr/local/cuda
jobs: jobs:
- TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cpu # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cpu
# - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu92 # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu92
# - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu100 # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu100
# - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu101 # - TORCH_VERSION=1.4.0 PYTHON_VERSION=3.8 IDX=cu101
......
script/cuda.sh
View file @ 27b934a8
...@@ -82,7 +82,11 @@ if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" != "cpu" ]; then ...@@ -82,7 +82,11 @@ if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" != "cpu" ]; then
PowerShell -Command "Start-Process -FilePath \"cuda_${CUDA}_win10\" -ArgumentList \"-s nvcc_${CUDA_SHORT}\" -Wait -NoNewWindow" PowerShell -Command "Start-Process -FilePath \"cuda_${CUDA}_win10\" -ArgumentList \"-s nvcc_${CUDA_SHORT}\" -Wait -NoNewWindow"
ls /c ls /c
ls /c/Program\ Files\ \(x86\) ls /c/Program\ Files\ \(x86\)
ls /c/Program\ Files\ \(x86\)/NVIDIA\ GPU\ Computing\ Toolkit # ls /c/Program\ Files\ \(x86\)/NVIDIA\ GPU\ Computing\ Toolkit
ls /c/Program\ Files\ \(x86\)/NVIDIA\ GPU\ Computing\ Toolkit/CUDA # ls /c/Program\ Files\ \(x86\)/NVIDIA\ GPU\ Computing\ Toolkit/CUDA
ls /c/Program\ Files\ \(x86\)/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v9.2 # ls /c/Program\ Files\ \(x86\)/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v9.2
ls /c/Program\ Files
# ls /c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit
# ls /c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA
# ls /c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v9.2
fi fi
script/rename_wheel.py
View file @ 27b934a8
...@@ -14,6 +14,6 @@ for wheel in wheels: ...@@ -14,6 +14,6 @@ for wheel in wheels:
continue continue
names = name.split('-') names = name.split('-')
name = '-'.join(names[:-4] + [names[-4] + '%2B' + idx] + names[-3:]) name = '-'.join(names[:-5] + [names[-5] + '%2B' + idx] + names[-4:])
new_wheel = osp.join(*wheel.split(osp.sep)[:-1], name) new_wheel = osp.join(*wheel.split(osp.sep)[:-1], name)
os.rename(wheel, new_wheel) os.rename(wheel, new_wheel)
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