Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/megatron-lm.git" did not exist on "a02a5490baab3b4745844b0f0752fe746a0cb7bc"
Commit 823a34e2 authored by rusty1s's avatar rusty1s
Browse files

update

parent b129b238
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 9 deletions
.github/workflows/building-conda.yml
View file @ 823a34e2
...@@ -14,17 +14,13 @@ jobs: ...@@ -14,17 +14,13 @@ jobs:
os: [ubuntu-18.04, macos-10.15] # windows-2019 os: [ubuntu-18.04, macos-10.15] # windows-2019
python-version: ['3.7', '3.8', '3.9', '3.10'] python-version: ['3.7', '3.8', '3.9', '3.10']
torch-version: [1.13.0, 2.0.0] torch-version: [1.13.0, 2.0.0]
# cuda-version: ['cpu', 'cu116', 'cu117', 'cu118'] # We have trouble building for `cu116` due to PyTorch 1.13.0 bugs
cuda-version: ['cu116'] cuda-version: ['cpu', 'cu117', 'cu118']
exclude: exclude:
- torch-version: 2.0.0 - torch-version: 2.0.0
python-version: '3.7' python-version: '3.7'
- torch-version: 1.13.0 - torch-version: 1.13.0
cuda-version: 'cu118' cuda-version: 'cu118'
- torch-version: 2.0.0
cuda-version: 'cu116'
- os: macos-10.15
cuda-version: 'cu116'
- os: macos-10.15 - os: macos-10.15
cuda-version: 'cu117' cuda-version: 'cu117'
- os: macos-10.15 - os: macos-10.15
......
.github/workflows/cuda/cu116-Linux-env.sh
View file @ 823a34e2
#!/bin/bash #!/bin/bash
export CUDA_HOME=/usr/local/cuda-11.6 CUDA_HOME=/usr/local/cuda-11.6
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
export PATH=${CUDA_HOME}/bin:${PATH} PATH=${CUDA_HOME}/bin:${PATH}
export FORCE_CUDA=1 export FORCE_CUDA=1
export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6" export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
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