输入：nvcc -V
输入：nvidia-smi
torch无法使用cuda
使用ubuntu时也出现同样的问题，最后发现是自己安装时命令没没搞明白。例如实验室服务器的cuda版本是10.1，所以在安装时先查看官网pytorch官网
以torch1.7.0为例，如果使用conda安装，则：
如果使用pip安装：
之前使用pip安装时一直没有加上“+cu101”，其中torchvision根据命令行提示，安装了torchvision+cu101版本。
OK
记录torch.cuda.is_available()==false；print(torch.__version__) ---＞1.8.0+cpu
猜你喜欢
转载自blog.csdn.net/weixin_44894550/article/details/121903011
今日推荐
周排行