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

This article mainly introduces the relevant knowledge of "how to view cuda version in python". Xiaobian shows you the operation process through actual cases. The operation method is simple, fast and practical. I hope this article "how to view cuda version in python" can help you solve the problem.

Specific operation method:

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

Open the idle tool in Python.

3. In the idle tool, click "File" in the upper left corner and select "New File" to create a shell script.

4. Import the torch module by entering the following command in a shell script:

import torch #Import torch module

5. Finally, check the cuda version through the following instructions.

print(torch.version.cuda)

Related operating instructions:

print(torch.cuda.is_available()) #See if cuda is available

print(torch.cuda.device_count()) #See the number of cuda available

About "how to view cuda version in python" content introduced here, thank you for reading. If you want to know more about industry-related knowledge, you can pay attention to the industry information channel. Xiaobian will update different knowledge points for you every day.

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.