Give it a try, Jeremy shows how to do it end-to-end on a notebook using PyTorch and provided simple instruction to make cuda enabled conda environment.
Jeremy demonstrates the entire process using PyTorch in a notebook and offers straightforward instructions for setting up a CUDA-enabled Conda costco ess environment.