NLP课程第一个实验,需要安装PyTorth,下面是一些安装方法及遇到的问题记录。
一、本地安装环境Windows10 + VScode
网上大多数都是 anaconda+vscode+pytorth 的环境组合配置,由于我采用本地安装配置,所以不需要用conda搭建虚拟环境。
直接进入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版本号;
(1)在官网中查看以前的版本;
(2)找到适配自己电脑的CUDA版本,按照官网提供的命令操作即可,例如:我的CUDA11.1 即可选择PyTorch1.9.0版本;
以上就是我的简单的PyTorch安装历程,如有错误请指正。



