栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Windows+pytorch深度学习框架安装

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Windows+pytorch深度学习框架安装

(liux系统ubuntu的可查看) 1.安装anaconda(清华镜像)

我选的这个anaconda3代表里面的是python3->然后一路next

在此处将第一位也打勾,可以自动为你添加路径,否则需要自己手动添加路径

2 .安装pytorch

选择好配置之后,复制命令

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

如果直接在源路径下下载,直接打开Anaconda powershell Prompt,先创建pytorch环境变量conda create -n pytorch,然后激活环境变量conda activate pytorch,然后输入复制的命令运行即可

但可能速度很慢,可以现在Anaconda powershell 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/

 创建pytorch环境变量conda create -n pytorch,然后激活环境变量conda activate pytorch, 删除安装指令中的-c pytorch,直接输入conda install pytorch torchvision torchaudio cudatoolkit=11.3运行。

在y/n?选择y 

输入python查看python环境

输入 import torch,如果未报错,则pytorch安装成功

输入torch.cuda.is_available(),如果显示Turn,则GPU可被pytorch使用

出现问题:

显示False

进入pyorch环境,conda list查看pytorch,如果发现序号带cpu说明安装错了,安装成cpu版本的了

conda uninstall pytorch卸载

在这里找pytorch,torchvision,torchaudio的gpu的包

 

cu113代表cuda 11.3版本的,cp39对应python3.9,windows就是win,liux就是liux。 torchvision和torchaudio也是同样的道理。下载下来

这里需要注意,下载的torch、torchvision和torchaudio版本一定要相互对应,具体可查看官网,这里的包都是有一定的依赖关系的。

 

 然后可以进入下载路径,即保存上述文件的路径中,cd +路径 (cd C:UsershpDownloads)

执行以下命令

//pip install 文件


pip install torch-1.10.0+cu113-cp39-cp39-win_amd64.whl
pip install torchvision-0.11.0+cu113-cp39-cp39-win_amd64.whl
pip install torchaudio-0.10.0+cu113-cp39-cp39-win_amd64.whl

让后再激活环境,一步步重新输入就是ture了

 重新查看conda list

显示GPU版本

3.安装pycharm

(在Ubuntu系统安装+深度学习框架_爱吃糖葫芦的大熊的博客-CSDN博客中也有讲解)

右下角配置环境,选择 interpreter settings

 选择show all

点击+号 

 选择conda已存在的环境,ok即可

 右下角就能选择该环境了

写个代码import torch, torch.cuda.is_available()试一下

 返回Ture

配置完成

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/856152.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号