Pytorch实战系列1——环境搭建(Anaconda+pytorch)

417 阅读2分钟

该搭建攻略是作者查阅了资料后总结的自己的理解及实现过程

借阅资料来自:blog.csdn.net/thomaswu199…

第一步 安装Anaconda

官网地址:www.anaconda.com/

根据需求下载相应版本的anaconda,安装过程中除了自定义安装路径外,其他的均next即可

第二步 新建anaconda单独的开发环境

(可以对于每一个项目都创建一个单独的开发环境,每开发一个新项目都会重新导入或者下载一些新的库)

1.打开anaconda navigator image.png

2.创建环境

root是python默认的环境,我们的目的是配置pytorch,那创建一个名为pytorch的环境即可 image.png

image.png

3.配置环境变量

我们再回到刚刚安装anaconda的路径,找到名为 envs文件夹并点开,找到我们新建的pytorch环境文件夹,点开新的环境:找到里面名为:Scripts的文件夹,此时复制这个文件夹的路径。

将该路劲粘贴进系统用户环境变量path中即可

image.png

第三步 安装配置pytorch

1.打开Anaconda Prompt

image.png

2.建立Pytorch Anaconda虚拟环境

#pytorch为环境名,这里创建python3.8版。

conda create -n pytorch python=3.8 anaconda
  • 其中加入最后一项anaconda命令建立虚拟环境时,会同时安装例如Notebook、Numpy、Scipy、Matplotlib、Pandas等数据分析包,用户也可以不加,选择后续安装。

3.启动Pytorch Anaconda虚拟环境

activate pytorch

4.安装pytorch

#安装方式如下:
#cpu版本
conda install pytorch-cpu torchvision-cpu -c pytorch
#GPU版
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
命令不可以正常下载则访问 pytorch.org/get-started…
根据需要的cuda版本,python版本进行下载

(需要注意的是,下载cuda版本时,需要查看自己的电脑是否具有gpu,如果没有gpu请下载cpu中央处理器版本)

查看电脑是否有gpu请看:blog.csdn.net/weixin_4200… image.png

5.测试pytorch

命令行输入

python  #进入python
import torch
import torchvision
print(torch.__version__)
exit()   # 退出python

6.关闭Pytorch Anaconda虚拟环境

conda deactivate

(对于下载了gpu版本的同学,只需在import torch之后,输入torch.cuda.is_available()查看输出即可)

tips

  • 如果需要使用本虚拟环境在Notebook中跑项目,进入工作目录激活虚拟环境,输入Jupyter Notebook运行或者直接运行anaconda jupyter notebook的小黑窗即可

  • 如果需要使用本虚拟环境在Pycharm进行项目开发,将设置里的Project Interpreter改为相应Anaconda文件目录下的Pytorch虚拟环境中的python.exe文件即可。(如: