Skip to content

Commit

Permalink
📝 update docs
Browse files Browse the repository at this point in the history
  • Loading branch information
AmineDjeghri committed Dec 17, 2023
1 parent 0ba9da7 commit b75dcde
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions Writerside/topics/unix_workflow/1_cuda_pytorch_install.md
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ One for GPU containing Pytorch with CUDA named `conda-env-gpu.yml` another one f
running the requirements.txt won't install again pytorch since the same version is installed.
- Running the conda-env-cpu.yaml that doesn't contain pytorch will install pytorch CPU from requirements.txt

Here is an example of a project containing a conda-env-gpu.yaml and a conda-env-cpu.yaml [link](package_example.)
Here is an example of a project containing a conda-env-gpu.yaml and a conda-env-cpu.yaml [link](package_example)
The files you mentioned are used for setting up different environments for your Python project.
Here's a brief summary of each:

Expand All @@ -83,7 +83,7 @@ The `-n my-env` option will supersede the environment name specified within the
- Run `nvcc --version; # should be cuda_11.8.r11.8`
- Verify the successful installation of PyTorch by executing the following Python code:

```py
```python
import torch
# setting device on GPU if available, else CPU
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
Expand Down
2 changes: 1 addition & 1 deletion unix_workflow/1_cuda_pytorch_install.md
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ One for GPU containing Pytorch with CUDA named `conda-env-gpu.yml` another one f
running the requirements.txt won't install again pytorch since the same version is installed.
- Running the conda-env-cpu.yaml that doesn't contain pytorch will install pytorch CPU from requirements.txt

Here is an example of a project containing a conda-env-gpu.yaml and a conda-env-cpu.yaml [link](package_example.)
Here is an example of a project containing a conda-env-gpu.yaml and a conda-env-cpu.yaml [link](package_example)
The files you mentioned are used for setting up different environments for your Python project.
Here's a brief summary of each:

Expand Down

0 comments on commit b75dcde

Please sign in to comment.