| name: vbeach | |
| channels: | |
| - conda-forge | |
| - pytorch | |
| - nvidia | |
| dependencies: | |
| - tqdm | |
| - pytorch | |
| - torchvision | |
| - torchaudio | |
| - pytorch-cuda=12.1 | |
| - scipy | |
| - numpy | |
| - matplotlib | |
| - pandas | |
| - scikit-learn | |
| - opencv | |
| - gxx_linux-64 | |
| - pip | |
| - pip: | |
| - torch-scatter -f https://data.pyg.org/whl/torch-2.1.2+cu121.html | |
| - segmentation-models-pytorch==0.2.0 | |