General custom tools to work in deep learning research using Python and PyTorch
This package can easily be installed using pip:
pip install pyvtorchAn alternative installation that partially uses Anaconda would involve...
- First, install some Anaconda distribution, in case you do not have any: https://docs.anaconda.com/anaconda/install/
- Then, create an Anaconda environment with Python
conda create -n dev python
- Activate the environment
conda activate dev
- Then, install all required packages by running the
install.shscript:yes | . install.sh
- You can make sure that your PyTorch installation has CUDA GPU support by running...
The first line should print
python -c "import torch; print(torch.cuda.is_available()) \ print([torch.cuda.get_device_name(i) for i in range(torch.cuda.device_count())])"
Trueif CUDA is supported. And the second line should show you the name/s of your available GPU/s. - That's it! You're good to go :)
That second installation procedure is designed to be overly redundant, so please feel free to follow your own installation procedure.
Provided installation steps are only guaranteed to work in Ubuntu 24.04 with NVidia drivers 535.
Valeria Pais - @vrpais - valeriarpais@gmail.com