Advertisement

在安装pytorch-cuda10.0时未指定pytorch及torchvision版本号会导致缺省安装问题需搭配正确版本

阅读量:

没加版本号命令

最开始安装cuda10.0对应的pytorch时候,命令如下:

复制代码
    conda install pytorch torchvision torchaudio cudatoolkit=10.0

正确命令:

复制代码
    conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0

可以看到值得注意的是,在项目中不仅未添加任何版本号信息(version字段),反而在依赖管理机制中增加了torchaudio库的引用(这可能是因为在某些环境中存在多个CUDA版本与PyTorch库对应的问题),并将其设置为固定值=10.0以规避潜在问题

在这里插入图片描述

默认安装了PyTorch 1.4.0以及 torchvision 0.5.0,
与官方提供的适配版本完全不同,
实际上,并非像我所预期那样能够智能地自动下载对应版本。
因此最好按照官方指导来操作。

复制代码
    conda install pytorch==1.2.0 torchvision==0.4.0

全部评论 (0)

还没有任何评论哟~