Commit 88b17ceb authored by rusty1s's avatar rusty1s
Browse files

[skip ci] update cuda installation

parent 8c84c156
...@@ -39,7 +39,7 @@ jobs: ...@@ -39,7 +39,7 @@ jobs:
- name: Install Cuda ${{ matrix.cuda-version }} - name: Install Cuda ${{ matrix.cuda-version }}
run: | run: |
./.github/workflows/cuda.sh source .github/workflows/cuda.sh
- name: Install PyTorch ${{ matrix.torch-version }}+${{ matrix.cuda-version }} - name: Install PyTorch ${{ matrix.torch-version }}+${{ matrix.cuda-version }}
run: | run: |
......
...@@ -18,4 +18,8 @@ sudo apt clean ...@@ -18,4 +18,8 @@ sudo apt clean
ls -lah /usr/local/ ls -lah /usr/local/
CUDA_HOME=/usr/local/cuda-11.1
LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
PATH=${CUDA_HOME}/bin:${PATH}
nvcc --version nvcc --version
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