win10从零安装配置pytorch全过程图文详解

所属分类: 脚本专栏 / python 阅读数: 1383
收藏 0 赞 0 分享

1.安装anaconda

(anaconda内置python在内的许多package,所以不用另外下载python)
可以点击下面的清华开源软件镜像站,在官网下载anaconda不如在这下的快
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 镜像站

我下载的是下图这个版本,对应的是3.7,anaconda3代表对应python3

我下载的是这个版本在这里插入图片描述

下载过程中基本上都是默认选项

在这里插入图片描述

在这里插入图片描述

在这里插入图片描述

在这里插入图片描述

下面这步可以不勾选add anaconda to my PATH environment variable,事后可以再添加

在这里插入图片描述

如果有下面这步,记得跳过,我这个版本的anaconda是没有这步的

在这里插入图片描述

在这里插入图片描述

下面检验一下安装成功了没有,在开始打开anaconda prompt,看到如图所示即成功

在这里插入图片描述

在这里插入图片描述

2.确认显卡及及时更新(为之后的pytorch安装作准备)

打开任务管理器-性能,查看有GPU还是只有集显

在这里插入图片描述

再win+r,运行cmd,输入nvidia-smi后回车

我这里发生如下报错,在环境变量path里面添加C:\Program Files\NVIDIA Corporation\NVSMI 即可解决

在这里插入图片描述

正常运行得到的

在这里插入图片描述

这里需要注意的是,表中的Driver Version需大于396.26,否则就去geforce experience更新显卡吧,我这里更新了最新版。

3.下载pytorch

https://pytorch.org/官网
这里需要注意的是,如果之前第二步查看时有GPU,则CUDA选择10即可,如果选9,之后不好用镜像源下载,速度会很慢;如果只有集显,那么CUDA选择None即可

在这里插入图片描述

之后我们复制Run this Command里的安装指令

在这里插入图片描述

打开anaconda prompt,输入下列指令:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/free/

既然选择了用镜像源下载,那么安装指令的 -c pytorch需删去,也就是:conda install pytorch torchvision cudatoolkit=10.1

这里我们创建虚拟环境conda create -n your_env_name(这里就是你给虚拟环境取的名字,笔者取名pytorch) python=X.X(笔者选下3.6),之后在里面操作,输入指令conda install pytorch torchvision cudatoolkit=10.1,然后y(es)

在这里插入图片描述
在这里插入图片描述

最后我们检验安装是否成功,先键入python进入python工作环境,然后键入import torch,如果未报错,则pytorch安装成功,之后输入指令torce.cuda.is_available(),如果显示Turn,则GPU可被pytorch使用

在这里插入图片描述

4.pycharm配置

打开pycharm,文件-新项目-Pure Python
Location自行选择,Interpreter选择打开你安装anaconda的位置,选择envs/your_env_name(虚拟环境名)/python.exe即可

在这里插入图片描述

创建后 在python console处运行再次运行import torch,无报错后运行torch.cuda.is_available(),如果显示Turn,则配置

更多精彩内容其他人还在看

Python环境管理virtualenv&virtualenvwrapper的配置详解

这篇文章主要介绍了Python环境管理virtualenv&virtualenvwrapper的配置详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
收藏 0 赞 0 分享

ITK 实现多张图像转成单个nii.gz或mha文件案例

这篇文章主要介绍了ITK 实现多张图像转成单个nii.gz或mha文件案例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
收藏 0 赞 0 分享

.img/.hdr格式转.nii格式的操作

这篇文章主要介绍了.img/.hdr格式转.nii格式的操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
收藏 0 赞 0 分享

python使用nibabel和sitk读取保存nii.gz文件实例

这篇文章主要介绍了python使用nibabel和sitk读取保存nii.gz文件实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
收藏 0 赞 0 分享

使用ITK-SNAP进行抠图操作并保存mask的实例

这篇文章主要介绍了使用ITK-SNAP进行抠图操作并保存mask的实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
收藏 0 赞 0 分享

基于python实现音乐播放器代码实例

这篇文章主要介绍了基于python实现音乐播放器代码实例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
收藏 0 赞 0 分享

Python 存取npy格式数据实例

这篇文章主要介绍了Python 存取npy格式数据实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
收藏 0 赞 0 分享

Python代码执行时间测量模块timeit用法解析

这篇文章主要介绍了Python代码执行时间测量模块timeit用法解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
收藏 0 赞 0 分享

在keras里实现自定义上采样层

这篇文章主要介绍了在keras里实现自定义上采样层,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
收藏 0 赞 0 分享

用Python开发app后端有优势吗

在本篇文章里小编给大家整理的是关于app后端开发学PHP还是Python的先关问题内容,需要的朋友们可以参考下。
收藏 0 赞 0 分享
查看更多