一、使用pip安装pytorch

(1)错误原因

在这里插入图片描述

(2)解决方案

  • 查看cuda版本
nvcc --version
  • 登陆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

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐