Hi All, sorry if I am late to the party.
evironment.yml first. It is best to install torch and torchvision from the torch channels (see https://pytorch.org/). For this, one can specify the channels that differ from
defaults by simply adding the channel before the package to be installed, e.g.
- pytorch::torch, so the environment definition will become:
Regarding the python version, I personally use
pytorch=1.7.1 with python 3.7 (
py3.7_cuda10.1.243_cudnn7.6.3_0 ). I think it is best to stick to that… @rrrrrok I did downgrade it once, but it is kinda messy as renku wants python 3.7 which is the one provided in the docker image. It was much simpler to use 3.7 or even 3.8, torch now provides versions for a variety of CUDA and python combos. And I never used that within a renkulab environment, it was a local env. I would just pin all package versions to make sure that all packages are happy with python 3.7.
@anisioti may I ask you what package must use python 3.6? Maybe I can help more!