寒假进度1PyTorch环境配置与安装

PyTorch环境配置与安装

学习背景

原本是打算学习区块链来参加服务外包杯的,但是在与专业老师的交流中,发现区块链对于本科的我们来说有些难度,所以小组讨论过后决定做音频多人声分离算法,因此来学习pytorch

学习链接来自B站:PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】_哔哩哔哩_bilibili

1.下载安装Anaconda

到官网(https://www.anaconda.com/)点击右上角,

选择下载

在新页面最底部选择版本

我选择的是python 3.9的64位,之后选择下载保存路径

双击进行安装,之后就是一些选择路径以及确定的简单操作,有疑问可以去本博客上方的视频教程中学习。

在最近添加中选择anaconda prompt,点击进入

若看到(base)则安装成功

2.配置与安装pytorch

在base页面输入conda create -n pytorch python=3.9(我python是3.9所以这里是python=3.9,如果你是3.6的python,这里就是python=3.6)

在页面输入y

下列中并没有pytorch,所以需要我们到官网下载

在官网 https://pytorch.org/ 中下载pytorch

以下是我个人的版本选择,仅供参考

复制页面中的命令到pytorch下的anaconda prompt

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

在他的安装列表中可以看到我们所需要的pytorch版本,输入y进行确认安装

检验安装:再次重复刚才的指令

之后进行以下命令,出现true则配置成功

 

3.扩展

在命令窗口查看自己的GPU版本,因为CUDA9.2版本以上只支持驱动版本大于396.26,在进行第二步的命令行安装前需要先检查一下自己的驱动版本

原文地址:https://www.cnblogs.com/wangdayang/p/15756326.html