Hi guys,
Can anyone please help me?
I am trying to run a .py
script for which I need an older pytorch version, because a function I
am using is deprecated in later torch versions.
But I seem not to be able to install it correctly.
I installed torch into my virtual environment using
conda create -n
my_env
python=3.6.2
source activate my_env
conda
install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.2 -c
pytorch
Then I have a python file (myfile.py) that I start from a shell file (start.sh).
start.sh
source activate my_env
srun --unbuffered python myfile.py
myfile.py
import torch
print(torch.__version__)
The print command from myfile.py still returns "1.8.0".
Am
I doing something very obvious wrong possibly? Did I install in a wrong
way, or is somehow my environment not properly loaded in the python
file?
Best regards and thanks a lot in advance
Dinho