Shulou(Shulou.com)05/31 Report--

This article mainly introduces "how to check whether cuda is available in python". In daily operation, I believe many people have doubts about how to check whether cuda is available in python. Xiaobian consulted all kinds of materials and sorted out simple and easy-to-use methods of operation. I hope it will be helpful to answer the doubt of "how to check whether cuda is available in python". Next, please follow the editor to study!

Specific operation methods:

1. Find the python program in the computer start menu.

2. Open the idle tool in python program.

3. Click "File" in the upper left corner of the idle tool and select "New File" to create a new shell script.

4. Enter the following instructions in the shell script to import the torch module.

Import torch # Import torch module

5. Finally, use the following instruction to check whether cuda is available.

Print (torch.cuda.is_available ())

Example of run result:

> > print (torch.cuda.is_available ())

True # returns True indicating that cuda is available

Related operation instructions:

Print (torch.cuda.device_count ()) # View the number of feasible cuda

At this point, the study on "how to check whether cuda is available in python" is over. I hope to be able to solve your doubts. The collocation of theory and practice can better help you learn, go and try it! If you want to continue to learn more related knowledge, please continue to follow the website, the editor will continue to work hard to bring you more practical articles!

Welcome to subscribe "Shulou Technology Information " to get latest news, interesting things and hot topics in the IT industry, and controls the hottest and latest Internet news, technology news and IT industry trends.