Skip to content

Commit

Permalink
Create McDevol_GPU.yml
Browse files Browse the repository at this point in the history
  • Loading branch information
yazhinia authored Dec 2, 2024
1 parent 6d78f00 commit 845bde1
Showing 1 changed file with 34 additions and 0 deletions.
34 changes: 34 additions & 0 deletions .github/workflows/main.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
name: McDevol_GPU

on: [push]

jobs:
build-linux:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- name: Set up Python 3.9
uses: actions/setup-python@v2
with:
python-version: 3.9
- name: Install CUDA Toolkit
uses: Jimver/[email protected]
id: cuda-toolkit
with:
cuda: '11.2.2'
- name: Create and activate conda environment
run: |
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh -b -p $HOME/miniconda
echo "$HOME/miniconda/bin" >> $GITHUB_PATH
source $HOME/miniconda/bin/activate
conda env create --file environment.yml --name mcdevol_env
conda activate mcdevol_env
- name: Install TensorFlow
run: |
pip install tensorflow-gpu==2.8.0
- name: Verify CUDA and TensorFlow
run: |
conda activate mcdevol_env
python mcdevol/mcdevol.py -h
python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

0 comments on commit 845bde1

Please sign in to comment.