今天使用conda安装PyTorch出现了错误,在此记录。
conda有了新的版本,让我更新,我先是没有管他。
然后进入PyTorch官网下载,速度极其慢,于是终止下载。
随后我更新了一下conda,最新版本4.8.4
又转用国内的清华开源镜像下载PyTorch,然后就发现标题的错误。
使用清华镜像需要设置下下载源,在Anaconda Prompt中输入以下

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/
  • 1
  • 2
  • 3
  • 4
  • 5

下面这个是查看下载源,上面命令执行后会你就会看到那些下载源的网址

conda config --show channels

 
 
  • 1

下面这个是切换回默认下载源,等你安装好PyTorch之后再用该命令

conda config --remove-key channels

 
 
  • 1

复制PyTorch官网你选择的conda安装命令,注意将 -c pytorch删掉,我的如下

conda install pytorch torchvision cudatoolkit=10.2

 
 
  • 1

如果你可以顺利安装那么恭喜你,但是我出现了标题的那个错误。对于这个错误网上有以下几种解决方法。

1.输入以下命令

# 1.
conda update -n base conda
# 2.
conda update --all

 
 
  • 1
  • 2
  • 3
  • 4

然后再conda install pytorch torchvision cudatoolkit=10.2安装试一下,这里就是在更新conda这个下载器(姑且这么称它),然后更新conda下的所有包,我试了以下,要更新的太多了,有些包下载一半就不下载了,跳到下一个下载,不知其原因。

2.输入以下命令

conda config --add channels conda-forge
conda config --set channel_priority flexible

 
 
  • 1
  • 2

3.改用pip安装

在PyTorch官网的选项里把conda改成pip就行啦,然后复制命令,执行一***意你复制的命令是从官网下载的,很慢。
可以改成以下命令试试,就是在原命令后面加上-i https://pypi.tuna.tsinghua.edu.cn/simple some-package

pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

 
 
  • 1

我用了下,感觉还是挺慢的,也许你们下了会比我快,算是个解决方案吧,本着对问题探究到底,我没有使用这个方法(也是慢的原因)

4.发现的问题

这里就是我自己找出的毛病啦
注意到上面的安装命令
cudatoolkit=10.2,我的电脑是有英伟达显卡的,所以想安装含有CUDA的PyTorch,但是我看了下自己的NVIDIA,版本信息是9.1,这是不是和10.2不匹配的原因呢,然后我看了下PyTorch官网历史版本,只有9.2,再往前翻,找到了9.0,既然有9.0,索性就将上面的命令改了一下cudatoolkit=9.0,然后安装成功了!!!

在这里插入图片描述

这里我仅仅是将cudatoolkit改成了9.0,其实你也可以试试升级下你的NVIDIA显卡驱动,具体方式看这个博客,就是去官网更新下,也很简单

https://blog.csdn.net/weixin_41762173/article/details/104261249

更新之后也许就可以直接用10.2了呢。

OK,鼓捣了一天,本来心情很烦躁,记录一下好多了,希望可以帮到大家。

版权声明:本文为tiaochewang219原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。

转载原文链接:https://www.freesion.com/article/83641174468/

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐