22.02.2024 Views

Daniel Voigt Godoy - Deep Learning with PyTorch Step-by-Step A Beginner’s Guide-leanpub

Create successful ePaper yourself

Turn your PDF publications into a flip-book with our unique Google optimized e-Paper software.

3. PyTorch

PyTorch is the coolest deep learning framework, just in case you skipped the

introduction.

It is "an open source machine learning framework that accelerates the path from

research prototyping to production deployment." [22] Sounds good, right? Well, I

probably don’t have to convince you at this point :-)

It is time to install the star of the show :-) We can go straight to the Start Locally

(https://pytorch.org/get-started/locally/) section of PyTorch’s website, and it will

automatically select the options that best suit your local environment, and it will

show you the command to run.

Figure S.3 - PyTorch’s Start Locally page

Some of these options are given:

• PyTorch Build: Always select a Stable version.

• Package: I am assuming you’re using Conda.

• Language: Obviously, Python.

So, two options remain: Your OS and CUDA.

"What is CUDA?" you ask. Environment | 13

Hooray! Your file is uploaded and ready to be published.

Saved successfully!

Ooh no, something went wrong!