pytorch 如何使用cuda

首先，在最上面放进去

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

其次，网络进入device

net.to(device)

最后，数据进入device

inputs, labels = inputs.to(device), labels.to(device)

以上在运行程序的时候都没有出现问题，但在输出的时候报错了

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same

解决方式：

images, labels = data
# 在输出的时候出现的问题，我应该把image也进到cuda里面，然后出来的是inputs，这时候再把
# inputs 输入到net里面，总是，要用cuda就全都用cuda，否则都不用
inputs, labels = images.to(device), labels.to(device)
outputs = net(inputs)

也就是说，在验证使用的时候，还是要把image进入到device里面，然后再进入net里面，总之，所有的要么都进GPU，要么都不进GPU

pytorch 如何使用cuda相关推荐

pytorch版本和cuda版本对应
pytorch版本和cuda版本对应 pytorch官网高版本的pytorch一般能兼容低版本cuda. 例如:你需要 1.7.0 的 pytorch,那么 cuda 只能 11.0 及以下.官方推 ...
pytorch出现RuntimeError: CUDA out of memory的问题解决方法
pytorch出现RuntimeError: CUDA out of memory解决方法将batch-size设置小些试试,如果不行,继续第二步多线程num_works设置较小,最差设为0
【RTX3060（暗夜精灵）等系列显卡正确配置Pytorch及其对于cuda和cudann版本的深度学习环境（避免由于版本不适应导致重装）】
提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档 RTX3060(暗夜精灵)等系列显卡正确配置Pytorch及其对于cuda和cudann版本的深度学习环境(避免由于版本不适应导致重装 ...
深度学习环境配置超详细教程【Anaconda+PyTorch(GPU版)+CUDA+cuDNN】
深度学习环境配置入门深度学习,首先要做的事情就是要搭建深度学习的环境.不管你是Windows用户,Mac用户还是Ubuntu用户,只要电脑配置允许,都可以做深度学习,毕竟Windows.Mac和Ub ...
安装MMDET全过程历时五天从anaconda安装到pytorch安装到CUDA配置到MIM安装到MMCV安装最后到MMDET安装
深度学习安装笔记(三)--安装MMDET全过程-- 历时五天从anaconda安装到pytorch安装到CUDA配置到MIM安装到MMCV安装最后到MMDET安装写在前面安装流程一.Anacon ...
pyTorch下载和cuda下载以及学习笔记
pytorch官方网站,cuda官方网站 CUDA下载:https://developer.nvidia.com/cuda-toolkit-archive CUDNN下载:https://develo ...
如何解决pytorch 编译时CUDA版本与运行时CUDA版本不对应
转载请注明: 仰望高端玩家的小清新 http://www.cnblogs.com/luruiyuan/ 如何解决pytorch 编译时CUDA版本与运行时CUDA版本不对应如果pytorch的编译时 ...
pytorch遇见RuntimeError: CUDA out of memory的解决
RuntimeError: CUDA out of memory 1.查看是否其他程序占用显存遇到此类错误后,对于py格式的文件来说,程序会进行终止,也就是当前程序占用的显存将会被释放.此时可用 w ...
pytorch 查看gpu cuda版本
import torchprint(torch.version.cuda) cuda版本查看: nvcc -V 9.0, V9.0.176 cudnn版本查看: python版本查看:pyt ...
安装Pytorch如何选择CUDA的版本
安装Pytorch时CUDA的选择 Nvidia CUDA 查看CUDA版本方法查看CUDA的驱动API版本查看CUDA的运行API版本查看官方CUDA的运行API版本方法1 方法2 查看An ...

pytorch 如何使用cuda

pytorch 如何使用cuda相关推荐

最新文章

热门文章