Can GPU acceleration pytorch work?

This article mainly introduces judgment pytorch Do you support GPU Accelerated operation , It has a good reference value , I hope that's helpful . If there is a mistake or not fully considered , Please grant me your advice . How to determine pytorch Do you support GPU Speed up ？ What you care about , Here comes the answer .

gpu Speed up pytorch Can I use it ？

Add this code ：

print torch.cuda.is_available()

True

Judgment complete ! Talk about pytorch How to view gpu Information ~

Why transfer data to GPU It's called .cuda instead of .gpu, It's like transferring data to CPU The method called is .cpu? This is because GPU The programming interface of is CUDA, And not all of them right now GPU All support CUDA, Only a fraction of Nvidia Of GPU To support .

PyTorch The future may support AMD Of GPU, and AMD GPU The programming interface of is OpenCL, therefore PyTorch And it's reserved for .cl Method , For future support AMD Waiting GPU.

torch.cuda.is_available()

cuda Is it available ;

torch.cuda.device_count()

return gpu Number ;

torch.cuda.get_device_name(0)

return gpu name , Device index defaults from 0 Start ;

torch.cuda.current_device()

Returns the current device index ;

The above is personal experience , I hope I can give you a reference .

Share some of my artificial intelligence learning materials for free , Including some AI Common framework actual combat video 、 Image recognition 、OpenCV、NLQ、 machine learning 、 Computer vision 、 Videos such as deep learning and neural network 、 Courseware source code 、 Famous essence resources at home and abroad 、AI Hot papers 、 Industry reports, etc .

For better systematic learning AI, I recommend that you collect one .

Here are some screenshots , Free download method is attached at the end of the article .

One 、 Artificial intelligence courses and projects

Two 、 Famous essence resources at home and abroad

3、 ... and 、 Collection of papers on artificial intelligence

Four 、 AI Industry Report

Learn Artificial Intelligence well , Read more , Do more , practice , If you want to improve your level , We must learn to settle down and learn slowly and systematically , Only in the end can we gain something .

Click on the business card below , Scan the code and download the information for free .