Technical requirements

For the implementation in this chapter, we will use a simple CNN as our model and use MNIST as our dataset. We will use PyTorch for illustration purposes. The main library dependencies for this chapter's code are as follows:

• torch>=1.8.1
• torchvision>=0.9.1
• cuda>=11.0
• NVIDIA driver >=450.119.03

It is mandatory to have the correct versions of these libraries installed before proceeding with the chapter.