安装pytorch遇到的一些坑

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引起的。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。