一、前言 最近在学习pytorch框架，记录一些涉及到的知识点，方便后续查找和学习。 二、主要内容 内容可能有些散，初次学习，可能把握不好知识之间的连贯性和整体性，后续适当调整。 关于CUDA的一些函数接口： torch.cuda.is_available() #查看系统GPU是否可以使用，经常用来判断是否装好gpu版的pytorch torch.cuda.current_device() #返回当前设备序号 torch.cuda.get_device_name(0) #返回第0号设备的name
