该搭建攻略是作者查阅了资料后总结的自己的理解及实现过程
借阅资料来自:blog.csdn.net/thomaswu199…
第一步 安装Anaconda
官网地址:www.anaconda.com/
根据需求下载相应版本的anaconda,安装过程中除了自定义安装路径外,其他的均next即可
第二步 新建anaconda单独的开发环境
(可以对于每一个项目都创建一个单独的开发环境,每开发一个新项目都会重新导入或者下载一些新的库)
1.打开anaconda navigator
2.创建环境
root是python默认的环境,我们的目的是配置pytorch,那创建一个名为pytorch的环境即可
3.配置环境变量
我们再回到刚刚安装anaconda的路径,找到名为 envs文件夹并点开,找到我们新建的pytorch环境文件夹,点开新的环境:找到里面名为:Scripts的文件夹,此时复制这个文件夹的路径。
将该路劲粘贴进系统用户环境变量path中即可
第三步 安装配置pytorch
1.打开Anaconda Prompt
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…
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文件即可。(如: