mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-09 11:32:07 +08:00
Update Distributed training.md
This commit is contained in:
@@ -88,8 +88,8 @@ module avail
|
|||||||
module load cuda-10.0
|
module load cuda-10.0
|
||||||
# ------------------------
|
# ------------------------
|
||||||
|
|
||||||
# make sure you've loaded the latest cuda version on your system
|
# make sure you've loaded a cuda version > 4.0 and < 7.0
|
||||||
module load gcc-9.1
|
module load gcc-6.1.0
|
||||||
|
|
||||||
$ pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
|
$ pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
|
||||||
```
|
```
|
||||||
|
|||||||
Reference in New Issue
Block a user