用conda下载安装pytorch-gpu版|torch.cuda.is_available()False|torch.cuda has no attribute
在linux服务器上安pytorch各种报错,看到帖子说要安cuda、安cuDNN还是不行,不过最后找了个简单办法安了两天终于安好了(;へ:)。首先,pytorch是有cpu和gpu版本的,如果使用gpu版本的话,要安装对应gpu版本,否则后续后报错(可以用conda list看一下自己是不是安成cpu的了)
原创
2021-11-30 16:01:54 ·
3054 阅读 ·
0 评论