美文网首页
安装pytorch遇到的一些坑

安装pytorch遇到的一些坑

作者: 此方病 | 来源:发表于2019-12-17 17:43 被阅读0次

tf 和 pytorch

安装pytorch的时候没有重新创建过虚拟环境,发现pytorch把我的tf1.10吞了:

解决方案:重新创建过虚拟环境分开安装

torch.cuda.is_available()返回false

我先安装了cuda=10.1的pytorch:
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
然后torch.cuda.is_available()返回false,搜了一下大概是驱动不匹配的问题。

于是我先查了一下版本,

driver version和CUDA version是匹配的鸭。然后试图折腾cuda驱动发现太麻烦,故而改打pytorch的主意。

我尝试换适配当下cuda10.0的pytorch版本:
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

在安装过程中,我发现pytorch,不,应该说conda,把我的cuda强行升为10.1了

联系上图就很清晰了,driver version匹配不上CUDA version,所以就安装回10.0甚至9.2的就ok。

成功

总结:
conda用起来需要谨慎,注意看会把你的哪些东西升级哪些降级。上述问题全是conda引起的。

相关文章

网友评论

      本文标题:安装pytorch遇到的一些坑

      本文链接:https://www.haomeiwen.com/subject/zicenctx.html