当前位置: 代码迷 >> 综合 >> 踩坑指南!pytorch1.0.1安装!又是猛男落泪的一天!
  详细解决方案

踩坑指南!pytorch1.0.1安装!又是猛男落泪的一天!

热度:21   发布时间:2024-02-26 15:15:08.0

git的一个源码在train的时候要求pytorch的版本降到1.3.0以下。第一次试了1.2.0,下载太慢了,加源后源中找不到1.2.0,太拉垮了,看网上好多人都在用1.0.1,于是就试了一下这个。

添加源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

安装命令:

conda install pytorch==1.0.1 torchvision==0.2.2 cudatoolkit=10.2 

装好后检查:

python>>>import torch

无报错则证明安装成功。但是我出现错误:

from torch._C
import * ImportError: DLL load failed: 找不到指定的程序。

太鸡肋啊啊啊啊,这个就是版本不匹配的问题,但是是哪个不匹配我真的花了很大的时间。
可能的解决方法:

更改python版本,pytorch1.0.1自带的python版本为3.6.2,直接conda install python就可安装到对应版本。
更改numpy版本,看到有网有说更改为: conda install numpy=1.17

检查:

print(torch.__version__)
1.0.1

但是我现在cuda貌似和pytorch版本不匹配,等我再研究一下。