Conda不同环境安装相同版本PyTorch是否重复下载及安装命令选择

问题1解答 #

Conda不会直接复用base环境中已安装的PyTorch运行文件，但会优先读取全局包缓存中的安装包：

• Conda默认会把所有环境下载过的安装包存放在全局公共缓存目录（默认路径一般是~/miniconda3/pkgs或~/anaconda3/pkgs），只要你要在virtualEnv中安装的PyTorch版本、构建标识、来源频道和base环境已下载的版本完全匹配，就不会重复下载，会直接把缓存中的安装包解压到virtualEnv的独立环境目录中。
• 两个环境的PyTorch安装后是完全隔离的，修改任意一个环境的PyTorch相关文件都不会影响另一个环境。

问题2解答 #

建议你使用和base环境完全一致的带-c pytorch参数的命令，原因如下：

• PyTorch官方维护的pytorch频道是PyTorch包的官方发布渠道，版本更新最及时，且不同组件（pytorch、torchvision、torchaudio、cudatoolkit）的版本匹配经过官方验证，不会出现依赖冲突或功能异常。
• 如果你使用不带频道参数的命令，Conda会从你本地配置的默认频道（一般是defaults或conda-forge）搜索包，这两个频道的PyTorch包是第三方编译的，大概率找不到1.9.0这类旧版本，就算找到也可能出现CUDA适配异常、功能缺失的问题。
• 如果你要精准安装1.9.0版本，建议在命令中补充版本号限制，完整命令如下：

conda install pytorch==1.9.0 torchvision torchaudio cudatoolkit=10.2 -c pytorch

内容的提问来源于stack exchange，提问作者Anurag Deyol