Pytorch从零开始其一:环境安装
环境安装
一:软件安装
1:安装Miniconda:
https://docs.conda.io/en/latest/miniconda.html
Miniconda比较轻便,运行快所以选择Miniconda不选Anaconda
如图所示:
2:安装Pycharm
https://www.jetbrains.com/pycharm/download/#section=windows
选择免费的社区版,满足使用需求
二:环境配置
安装好两个软件,打开Anaconda prompt
1. 首先切换下载源为清华镜像,这样下载速度快一点:
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/
2. 下载源配置完毕后开始配置环境。
conda的环境相当于将所有运行的库装在一个袋子中,每个袋子互不干扰。方便不同版本的库的使用。
输入:
conda create -n pytorch python=3.6
之后全部选y,环境设置完毕
然后输入以下代码来启动环境:
conda avtivate pytorch
到这里,袋子创建完毕了,接下来就该向袋子中填充物品——库文件。
3. 安装pytorch所需的库
https://pytorch.org/
选择如图所示,红框即为下载所需代码:
在anaconda prompt中输入:
conda install pytorch torchvision torchaudio cudatoolkit=11.3
来安装库文件,cuda版本参考个人GPU型号,本人为GTX1060。
安装库文件时两点需要注意:
1,需切换清华的镜像源,否则下载速度极慢,容易断开连接
2,需在创建的pytorch环境中安装库
三:Pycharm导入环境
1.安装完Miniconda之后如图所示设置
新建项目选择先前配置的解析器,然后点击. . .
选择conda环境
选择Miniconda安装文件的目录,接着路径如下:
根目录\envs\pytorch\pythonexe
注意:这里的pytorch是环境的名字
2.测试
打开Python控制台,输入代码:
import torch
torch.cuda.is_available()
显示如图,表示安装成功