用anaconda配置pytorch环境

我觉得有必要写一篇博客记录我的辛酸历程

第一步 安装anaconda 在清华镜像源找个安装包

安装完是这个样子

anaconda Navigator是一个可视化软件

点开有当前的(base)环境

自己创建一个

直接用Anaconda Prompt命令行搞

先把下面的镜像装上

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 --show

里面有

镜像添加完就可以直接创建一个环境

conda create -n learn python=3.6

3.6版本的最稳定,不要装太高的,pytorch支持不了太高的

创建完之后进入环境

activate learn

你会发现(base)变成了(learn)

进去之后查看都有啥

conda list

然后就开始重中之中装pytorch

先去pytorch官网

https://pytorch.org/

找到这个图片 按我这么选,最后一行是命令,但是先别复制

倒数第二行行的cuda是显卡驱动, none是不装gpu的

看你是那个版本,点开nvdia控制面板

点这个

这里能看到你是啥版本,我的是11.1.106的 所有比这个小的cuda版本都能装

我装的11.1的cuda

conda install pytorch torchvision torchaudio cudatoolkit=11.0

注意后面的-c pytorch不要写!!

然后就等着安装吧,安装完了进anaconda Navigator看看

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐