写文章
- @weixin_46349825
简介
该用户还未填写简介
擅长的技术栈
未填写擅长的技术栈
可提供的服务
暂无可提供的服务
pytorch安装gpu版本但实际结果为cpu版本的可能错误
发现清华园的pytorch包列表里的11.2版本对应的都是cpu版本,所以conda命令就只给你下载与之相对的cpu版本。print(torch.cuda.get_device_name(0))# 显示第一个GPU的名字。那么就想到我们可以通过指定清华园里存在的pytorch包名称的cuda版本来实现指定安装。本着向下兼容的原则,我这里安装cudnn8.0,cuda11.1版本的pytorch。安
pytorch安装gpu版本但实际结果为cpu版本的可能错误
发现清华园的pytorch包列表里的11.2版本对应的都是cpu版本,所以conda命令就只给你下载与之相对的cpu版本。print(torch.cuda.get_device_name(0))# 显示第一个GPU的名字。那么就想到我们可以通过指定清华园里存在的pytorch包名称的cuda版本来实现指定安装。本着向下兼容的原则,我这里安装cudnn8.0,cuda11.1版本的pytorch。安
pytorch安装gpu版本但实际结果为cpu版本的可能错误
发现清华园的pytorch包列表里的11.2版本对应的都是cpu版本,所以conda命令就只给你下载与之相对的cpu版本。print(torch.cuda.get_device_name(0))# 显示第一个GPU的名字。那么就想到我们可以通过指定清华园里存在的pytorch包名称的cuda版本来实现指定安装。本着向下兼容的原则,我这里安装cudnn8.0,cuda11.1版本的pytorch。安
到底了