一、使用pip安装pytorch

(1)错误原因

在这里插入图片描述

(2)解决方案

  • 查看cuda版本
nvidia-smi
  • 登陆pytorch的官方网站,使用官方的下载语句下载即可。

官方网站

https://pytorch.org/

在这里插入图片描述

(3)安装完成的情形

在这里插入图片描述

二、使用cuda安装pytorch

首先添加镜像源:

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://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
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 --set show_channel_urls yes
  • 此时,目录 C:\Users<你的用户名> 下就会生成配置文件.condarc,将里边-defaults删除,内容为:
channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
show_channel_urls: true
ssl_verify: true
channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main

show_channel_urls: true
  • 进入pytorch官网
    在这里插入图片描述
  • 打开CMD窗口,在命令行中键入上述命令,等待安装成功(一定要注意:删除后边的-c pytorch,不然自己设置的conda源是不会起效果的,因为会默认去pytorch的官网下载)
conda install pytorch torchvision cudatoolkit=10.2
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

三、安装pytorch==1.2.0 和cuda 10.0对应

pip install torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
  • 安装pytorch==1.0.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda100 -c pytorch

参考

Windows下使用conda安装pytorch速度非常慢的解决方法

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐