VS code makes debugging our code and inspecting our objects pretty easy. We'll be using VS Code primarily for debugging our code. Once you have Visual Studio Code installed, you'll also want to install the Python plugin. In this series, we'll be using the following software for writing, debugging our code: # packages in environment at C:\Users\deeplizard\Anaconda3: # Name Version Build Channel All we need to do is select a version of CUDA if we have a supported Nvidia GPU on our system.
The needed CUDA software comes installed with PyTorch if a CUDA version Also, there is no need to install CUDA separately.
Notice that we are installing both PyTorch and torchvision. In this case, we have the following command:Ĭonda install pytorch torchvision cudatoolkit= 10.2 -c pytorch
