conda安装gpu版pytorch后cuda调用失败
问题:按pytorch官方教程安装gpu版pytorch(conda install pytorch torchvision cudatoolkit=10.2)后,在python环境下执行print(torch.cuda.is_available()),返回false。解决: conda卸载pytorchconda uninstall pytorch pip安装gpu版pytorch:pip install torch torchvision -...
原创
2020-08-13 22:20:45 ·
3505 阅读 ·
1 评论