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

PyTorch安装与环境配置

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

PyTorch安装与环境配置

PyTorch安装与环境配置

NLP课程第一个实验,需要安装PyTorth,下面是一些安装方法及遇到的问题记录。

一、本地安装环境

Windows10 + VScode
网上大多数都是 anaconda+vscode+pytorth 的环境组合配置,由于我采用本地安装配置,所以不需要用conda搭建虚拟环境。

二、安装PyTorch

直接进入PyTorch官网,按照自己电脑的配置选择相应的PyTorch版本。

官网首页提供的是最新版本,直接选择Stable根据操作系统选择Linux/Mac/WindowsConda搭建虚拟环境选择Conda,我是本地安装选择Pip计算系统选择 CPU 或者 CUDA(如果有navida的GPU的显卡),由于我的电脑配置是AMD系统,CUDA11.1,所以直接选择CUDA11.3即可。(按理说CUDA11.1与CUDA11.3应该是有区别的,但是装了适用CUDA11.3的PyTorch也没有什么问题,后面会介绍如何找到适配CUDA其他版本的Pytorch方法。)最后复制下面的 pip 命令至vscode终端即可。

如果想装适配CUDA其他版本的Pytorch,可以继续下面的内容。

三、如何查看CUDA版本

(1)搜索并打开NVIDIA控制面板;
(2)点击左下角的“系统信息”;
(3)在“组件”中可以查看到CUDA版本号;

四、如何安装其他CUDA版本的PyTorch

(1)在官网中查看以前的版本;

(2)找到适配自己电脑的CUDA版本,按照官网提供的命令操作即可,例如:我的CUDA11.1 即可选择PyTorch1.9.0版本;

五、总结

以上就是我的简单的PyTorch安装历程,如有错误请指正。

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

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

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