Pytorch下载安装配置

安装看这个博客即可,写的很清楚

https://www.cnblogs.com/zhouzhiyao/p/11784055.html

测试是否安装成功

https://blog.csdn.net/weixin_38314865/article/details/102410371
import torch
 
torch.cuda.is_available()
True

torch.cuda.current_device()
0

torch.cuda.device(0)
<torch.cuda.device object at 0x0000023034332438>
 
torch.cuda.device_count()
1
 
torch.cuda.get_device_name(0)
'GeForce GTX 1660'
©️2020 CSDN 皮肤主题: 数字20 设计师:CSDN官方博客 返回首页