conda配置pytorch,cuda等小技巧

用anaconda安装torch。新建虚拟环境后,直接在pytorch官网官网链接找到“Install”按钮。这里一键搞定torch,torchvision,cudatoolkit等等,不需要另外安装cuda,并且版本都会自己对于对应好,原因是anaconda都集成在虚拟环境里面了,统一管理各个依赖包。
conda配置pytorch,cuda等小技巧_第1张图片
在这里插入图片描述

如果不是用anaconda来管理环境,就要仔细对照torch、torchvision、cuda之间的对应关系。

搞了好久,泪目,end :)

你可能感兴趣的:(conda,ubuntu,1024程序员节)