您好,欢迎来到好走旅游网。
搜索
您的当前位置:首页利用清华镜像在Anaconda环境中安装pytorch

利用清华镜像在Anaconda环境中安装pytorch

来源:好走旅游网
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. 可以试试使用conda安装/升级pip
conda install pip
  1. 查看当前环境
conda info --envs

注:本文安装在默认环境【base】中,如有需要可以新建其他环境并切换至该环境,代码如下:

conda create -n pytorch python=3.8
conda activate pytorch
conda info --envs

即可看到 conda 环境中,有新建的 pytorch 环境,右边的 * 号表示,当前你处于哪个环境。

  1. 在Pytorch网站查询需要安装的版本(),如下:
  2. 复制安装代码,注意去掉后缀,如下。
conda install pytorch torchvision torchaudio cudatoolkit=10.2
  1. 弹出提示,输入 y,即可完成安装,显示“done”。

  2. 测试是否安装成功:

    (1)输入:conda.bat activate pytorch (不报错)

    (2)输入:python (显示版本号,如Python 3.7.7)

    (3)输入:import torch(不报错)

    (4)输入:torch.cuda.is_available(显示<function is_available … …>)

    完成以上(1)至(4),说明安装成功。

如果本文对你有用,请点个赞加个关注吧~

因篇幅问题不能全部显示,请点此查看更多更全内容

Copyright © 2019- haog.cn 版权所有

违法及侵权请联系:TEL:199 1889 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务