davrot/pytutorial
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update and rename pytorch/layers/cuda/README.md to pytorch/cuda/README.md

Browse source 
Signed-off-by: David Rotermund <54365609+davrot@users.noreply.github.com>
This commit is contained in:
David Rotermund 2023-12-08 14:47:47 +01:00 committed by GitHub
parent 0f69bd0f75
commit d75f6fb94a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
pytorch/layers/cuda/README.md → pytorch/cuda/README.md
View file

@ -8,15 +8,17 @@
## The goal
Convince PyTorch and Nvidia's GPUs working together. **I assume you installed the PyTorch and/or TensorFlow version for CUDA. (see Python installation instructions on thsi site...)**
Convince PyTorch and Nvidia's GPUs working together. **I assume you installed the PyTorch and/or TensorFlow version for CUDA. (see Python installation instructions on this site...)**
Questions to [David Rotermund](mailto:davrot@uni-bremen.de)
## Windows
## Windows / Linux
* Download and install [CUDA driver](https://developer.nvidia.com/cuda-downloads)
* Download and install [cuDNN toolkit](https://developer.nvidia.com/rdp/cudnn-download) (you will need to create an account :-( )
For Linux use a repo if possible. Windows and Linux require that you have install working graphic driver beforehand. At least that is what I believe is a requirement.
## Test the PyTorch
```python