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
联系上图就很清晰了,driver version匹配不上CUDA version,所以就安装回10.0甚至9.2的就ok。
总结:
conda用起来需要谨慎,注意看会把你的哪些东西升级哪些降级。上述问题全是conda引起的。