Python code for testing correct installation of dependencies for Pytorch cifar10 training.
-
Install Python
sudo apt install python
pip3sudo apt install -y python3-pip
and venvsudo apt install -y python3.12-venv -
Create a venv
python3 -m venv env_name -
Navigate to the directory
cd env_name -
Activate the venv
source bin/activate -
Install PyTorch by following https://pytorch.org/get-started/locally/ You should use the latest CUDA version supported by your system (likely just the latest available). This will look something like the following:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
- Run
python3 cifartest.pyand see if there are any packages missing and/or cuda is available.