torch.__version__:打印已安装的 PyTorch 版本。 torch.version.cuda:打印 PyTorch 所使用的 CUDA 版本。 torch.cuda.is_available():检查系统上是否有可用的 CUDA。 torch.cuda.device_count():计算可用的具有 CUDA 功能的设备数量。 CUDA 设备详情: torch.cuda.get_device_name(i):获取指定索引 i 处的 CUD...
链接:https://pan.baidu.com/s/1dZZTpN5wQjKv4iHXt3CxXg 提取码:5200 一、安装cuda 下好安装包之后有两个文件,一个是cuda安装包,形式如下: 另外一个是cudnn解压包,cudnn解压包要解压,解压之后有如下文件: 直接双击cuda安装包,会加载安装程序,然后点ok 前面可以默认,一直到安装选项,选择自定义,然后下一步...