Pytorch详细安装教程
- 1、安装Anacoda
- 2、在Anacoda环境中新建一个pytorch环境
- 3、按照官网的方法安装pytorch
- 4、测试安装
- 5、在Pytorch中安装jupyetr notebook
up在试了网上很多教程之后,一次次的都错,真的是安装了无数遍GPU版本的pytorch,使用清华源镜像但是有个巨坑,查了很多博客,终于有了点头绪顺利解决安装问题速度飞快。
1、安装Anacoda
链接: Anacoda下载地址. 具体的安装过程,网上一堆教程,自行查看 记得配置环境变量
2、在Anacoda环境中新建一个pytorch环境
至于为什么要新建一个环境,为了方便我们管理各个环境,就跟你自己的D盘好好分类一样,有时候的需要需要用到tensorflow 1.x的版本 ,有是有要用tensorflow2.x的版本,有是有需要pytorh的版本,多以新建一个环境很重要。下面来看看创建一个新环境吧: 1、打开Anconda prompt 进入普通的base环境
代码语言:javascript复制conda creat -n pytorch python=3.7
-n 后面紧跟的是环境的名称 需要依赖python 3.7的环境
2、打开创建的这个环境
代码语言:javascript复制activate pytorch
3、现在已经进入这个新的环境内,先自己配置一下清华镜像源,在环境中中运行以下代码
代码语言:javascript复制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/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/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
依次在pytorch的环境中输入完之后,打开C盘下的user下的username文件夹下的.condarc文件,用记事本打开,将https改为http,还有就是删掉-default,点击保存。
弄完上述之后,就可以安装pytorch gpu版本了,是不是很激动,稍等,对了这个gpu你的电脑支不支持还得看你英伟达的显卡,我的是gtx1660ti是没问题,一般的gpu都支持,至于安装的是cuda什么版本的还真的要去看你的英伟达显卡的型号了,怎么查看自己的英伟达显卡信息呢? 1、打开NVIDIA控制面板。
在下面所以我安装的10.2的,以下的版本都可以安装。如果你的呆脑没有英伟达显卡的就直接None安装CPU版本的吧。
3、按照官网的方法安装pytorch
链接: Pytorch下载地址.
尽量选择stable稳定版本,我选择的是上面的windows版本,conda安装,Python语言,CUDA版本是10.2 然后复制 run this command 上面的网址:
代码语言:javascript复制conda install pytorch torchvision cudatoolkit=10.2
至于为什么后面没有加入了这个 -c pytorch 的原因:其中,-c pytorch参数指定了conda获取pytorch的channel,在此指定为conda自带的pytorch仓库。 因此,只需要将-c pytorch语句去掉,就可以使用清华镜像源快速安装pytorch了。因为我们要使用清华镜像源进行安装(清华镜像源的配置如上第2点) 这样安装速度真的超快》》》》
整个过程不超过五分钟,原来直接安装,失败就算了,还次次都失败,气
4、测试安装
在刚刚创建的pytorch环境命令行中输入python 进入python 测试》》》》》》
代码语言:javascript复制import torch
如果没有报错说明安装成功 心在看看CUDA安装是否成功
代码语言:javascript复制torch.cuda.is_available()
返回为True则安装成功,大功告成,完美
5、在Pytorch中安装jupyetr notebook
jupyter 是一个交互特别好的界面,适合初学者学代码,怎么安装呢?看步骤 1、进入pytorch环境(activate pytorch)输入以下代码
代码语言:javascript复制conda install nb_conda
遇到选择的就是yes
之前配置好清华镜像源了,速度还是很快的 2、打开jupyter ,直接在命令行输入jupyter notebook 即可
进入之后,选择这个pytorch的环境即可
不报错说明成功!!
Pycharm的配置也很简单