You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+15-2Lines changed: 15 additions & 2 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -1,5 +1,5 @@
1
1
# Project Templete to Kickoff Machine Learning Project
2
-
> This template currently requires`Python 3.11` installed (`conda`) or be a default Python version (`poetry`)
2
+
> This template currently uses`Python 3.11`, and either `conda`, `docker`, `poetry`, or `micromamba`
3
3
4
4
Template project aims to promote *versioning library*, *environment isolation* practice and help all ML practitioners quickly start a project. Using this template, practitioners will have below libraries
5
5
+ Pytorch
@@ -17,7 +17,7 @@ Template project aims to promote *versioning library*, *environment isolation* p
17
17
18
18
Those libraries of course aren't enough, but it's easy to update other libraries that support your project.
19
19
20
-
> Using `poetry` is highly recommended.
20
+
> Using `poetry` is highly recommended. If you are using `conda` or `micromamba`, make sure that you use package hashes to ensure package selection is reproducible via `conda-lock` or `micromamba`
- This tutorial is for those who have NVIDIA GPU (hereafter GPU). For CPU case, this should be similar but need to adjust `Dockerfile` and `run_docker.sh`
65
+
- Note: Docker will use `micromamba` instead of `miniconda`. Replace `conda` with `micromamba` in your usual commands
66
+
- You need to install `docker`
67
+
- Install Nvidia driver (ignore if you don't have GPU)
68
+
- Then install [Nvidia docker container toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/index.html), ignore if you aim to use CPU.
69
+
- Edit `.env` locates the same level with `run_docker.sh`, to add environment variables to the prospective docker container
70
+
- There is a file named `run_docker.sh`, allow to execute it by `chmod +x run_docker.sh` and run `run_docker.sh`
71
+
- Enjoy Jupyter lab at localhost:8888 as usual
72
+
63
73
## After installing libraries, verify
64
74
try this in `ipython`
65
75
```python
@@ -88,3 +98,6 @@ If you aim to follow `conda`
88
98
- edit file `environment.yml`
89
99
- create a new environment that has the version you want
90
100
- switch to that environment
101
+
102
+
## Current limitation
103
+
-`torch_geometric` with CPU version in `poetry` has a problem. I created a discussion at https://github.com/pyg-team/pytorch_geometric/discussions/7788
0 commit comments