dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update TORCH_CUDA_VERSION usage (#1284)

This commit is contained in:
Guillaume Lagrange 2024-02-10 12:01:45 -05:00 committed by GitHub
parent 88f5a3e88c
commit 8f753b3aae
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
burn-tensor/README.md
View File

@ -50,7 +50,7 @@ Therefore, creating the tape only requires a simple and efficient graph traversa
## Cuda
To run with CUDA set `TORCH_CUDA_VERSION=cu113`.
To run with CUDA set `TORCH_CUDA_VERSION=cu121`.
## Notes

2
examples/mnist/README.md
View File

@ -17,7 +17,7 @@ cargo run --example mnist --release --features ndarray # CPU NdAr
cargo run --example mnist --release --features ndarray-blas-openblas # CPU NdArray Backend - f32 - blas with openblas
cargo run --example mnist --release --features ndarray-blas-netlib # CPU NdArray Backend - f32 - blas with netlib
echo "Using tch backend"
export TORCH_CUDA_VERSION=cu113 # Set the cuda version
export TORCH_CUDA_VERSION=cu121 # Set the cuda version
cargo run --example mnist --release --features tch-gpu # GPU Tch Backend - f32
cargo run --example mnist --release --features tch-cpu # CPU Tch Backend - f32
echo "Using wgpu backend"

2
examples/simple-regression/README.md
View File

@ -23,7 +23,7 @@ cargo run --example regression --release --features ndarray # CPU
cargo run --example regression --release --features ndarray-blas-openblas # CPU NdArray Backend - f32 - blas with openblas
cargo run --example regression --release --features ndarray-blas-netlib # CPU NdArray Backend - f32 - blas with netlib
echo "Using tch backend"
export TORCH_CUDA_VERSION=cu113 # Set the cuda version
export TORCH_CUDA_VERSION=cu121 # Set the cuda version
cargo run --example regression --release --features tch-gpu # GPU Tch Backend - f32
cargo run --example regression --release --features tch-cpu # CPU Tch Backend - f32
echo "Using wgpu backend"

2
examples/text-classification/README.md
View File

@ -29,7 +29,7 @@ cd burn
# Use the --release flag to really speed up training.
# Use the f16 feature if your CUDA device supports FP16 (half precision) operations. May not work well on every device.
export TORCH_CUDA_VERSION=cu117 # Set the cuda version (CUDA users)
export TORCH_CUDA_VERSION=cu121 # Set the cuda version (CUDA users)
# AG News
cargo run --example ag-news-train --release --features tch-gpu # Train on the ag news dataset

2
examples/text-generation/README.md
View File

@ -14,7 +14,7 @@ git clone https://github.com/tracel-ai/burn.git
cd burn
# Use the --release flag to really speed up training.
export TORCH_CUDA_VERSION=cu113
export TORCH_CUDA_VERSION=cu121
cargo run --example text-generation --release
```