Update start.sh
Browse files
start.sh
CHANGED
|
@@ -34,6 +34,9 @@ export LTXV_DEBUG=1
|
|
| 34 |
|
| 35 |
nvcc -V
|
| 36 |
|
|
|
|
|
|
|
|
|
|
| 37 |
python -c "import torch; print('PyTorch CUDA disponível:', torch.cuda.is_available()); print('Dispositivos:', torch.cuda.device_count())"
|
| 38 |
|
| 39 |
|
|
|
|
| 34 |
|
| 35 |
nvcc -V
|
| 36 |
|
| 37 |
+
pip install --extra-index-url=https://pypi.ngc.nvidia.com --trusted-host pypi.ngc.nvidia.com nvidia-tensorflow
|
| 38 |
+
|
| 39 |
+
|
| 40 |
python -c "import torch; print('PyTorch CUDA disponível:', torch.cuda.is_available()); print('Dispositivos:', torch.cuda.device_count())"
|
| 41 |
|
| 42 |
|