pytorch环境搭建 (windows环境)
Anaconda安装
登陆Anaconda官网进行下载并且安装,记录下安装路径,稍后在编辑器打开时选择该路径
在Anaconda中创建pytorch的运行环境
在conda中可以创建多个运行环境或者说工作环境,不同环境中可以安装不同版本的软件/工具,避免了直接在主机安装时智能存在一个版本的情况
1、创建要安装pytorch的环境
conda create -n <environment_name> python=<version>
例如我创建的名为 pytorch 使用python3.6的创建方法如下
conda create -n pytorch python=3.6
2、激活环境
conda activate <environment_name>
例如要激活刚刚创建的名为 pytorch 的环境,方法如下
conda activate pytorch
3、 在环境中安装pytorch
1)登陆pytorch官网,查询适用自己系统类型/显卡的安装命令
2)在刚激活的conda环境中输入查询到的命令进行安准过
4、验证安装是否成功
进入python交互模式 import torch
5、验证pytorch是否可以是用GPU
进入conda中创建的pytorch环境torch.cuda.is_available()
返回true代表正常使用GPU
查看GPU是否支持CUDA
查看GPU驱动版本
nvidia-smi
安装pytorch时推荐选择CUDA9.2,驱动版本支持CUDA9.2即可,即驱动版本要大于369.26,不满足时下载英伟达最新驱动进行更新。
下载编辑器Pycharm
登陆官网下载pycharm。如果是用jupyter,在安装conda的时候已经自带了jupyter,在开始中找到即可。
pycharm下载完成后,是用刚创建好并安装torch的环境pytorch。