个人记录torch.cuda.is_available()==false;print(torch.__version__) --->1.8.0+cpu

个人记录torch.cuda.is_available()==false;print(torch.__version__) --->1.8.0+cpu_第1张图片
输入:nvcc -V
在这里插入图片描述

输入:nvidia-smi
个人记录torch.cuda.is_available()==false;print(torch.__version__) --->1.8.0+cpu_第2张图片
torch无法使用cuda
使用ubuntu时也出现同样的问题,最后发现是自己安装时命令没没搞明白。例如实验室服务器的cuda版本是10.1,所以在安装时先查看官网pytorch官网
以torch1.7.0为例,如果使用conda安装,则:个人记录torch.cuda.is_available()==false;print(torch.__version__) --->1.8.0+cpu_第3张图片

如果使用pip安装:
个人记录torch.cuda.is_available()==false;print(torch.__version__) --->1.8.0+cpu_第4张图片
之前使用pip安装时一直没有加上“+cu101”,其中torchvision根据命令行提示,安装了torchvision+cu101版本。
OK

你可能感兴趣的:(笔记,pytorch,深度学习)