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

最新推荐文章于 2023-04-07 14:34:02 发布

cxm 17

最新推荐文章于 2023-04-07 14:34:02 发布

阅读量3.8k

点赞数 4

CC 4.0 BY-SA版权

文章标签：环境配置 pytorch

本文链接：https://blog.youkuaiyun.com/weixin_45415853/article/details/96196835

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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:/