How can I update to the latest (or at least a more recent) version of PyTorch with Lambda Stack? On 2022.02.10 PyTorch is up to 1.10.2 and CUDA up to 11.3 per https://pytorch.org/get-started/locally/.
I’ve a Lambda Labs Tensorbook. I tried upgrading my lambda stack with:
sudo apt-get update
sudo apt-get dist-upgrade
That succeeded. However
python3 -c "import torch;print(torch.__version__);print(torch.cuda.is_available());print(torch.cuda.get_device_name(torch.cuda.current_device()));"
outputs
1.6.0
True
NVIDIA GeForce RTX 2080 Super with Max-Q Design
which reveals the lambda-stack is using PyTorch 1.6.
Additionally
nvcc --version
outputs
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:24:38_PDT_2019
Cuda compilation tools, release 10.2, V10.2.89
which shows CUDA 10.2 in use.
Please help. I at least need Pytorch 1.8.