conda虚拟环境配置旧版pytorch0.4.1

最新推荐文章于 2023-10-06 16:01:05 发布

原创

最新推荐文章于 2023-10-06 16:01:05 发布 · 3.9k 阅读

标签

#环境配置 #pytorch

收录于

在PyTorch更新至1.0稳定版后，安装0.4.1版本遇到问题。通过修改conda源，使用conda命令安装，但初始安装的cudatoolkit未正确配置。解决方法是手动安装cuda8.0，这将自动调整相关模块。随后，分别安装torch和torchvision，最终成功完成旧版PyTorch环境配置。

Linux安装pytorch0.4.1

修改conda源
用conda命令安装
安装cuda8.0
安装torchvision

pytorch更新到1.0稳定版后，用官网命令安装旧版遇到了两个问题：

用pip install .whl安装后，conda list 只有torch 0.4.1, 没有pytorch 0.4.1
用conda命令安装，cudatoolkit没有自适应配置，导致torch.cuda.is_available()=False.

安装环境：Linux python3.6.5 cuda80 cudnn6

修改conda源

清华镜像源已经不支持pytorch了，不过一些基础库还在。

#添加第三方conda源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https:/