【pytorch】(pip install pytorch)安装失败,使用pip和conda安装pytorch
一、错误原因二、解决方案登陆pytorch的官方网站,使用官方的下载语句下载即可。官方网站https://pytorch.org/三、安装完成的情形
·
文章目录
一、使用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
参考
更多推荐
所有评论(0)