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,在进行第二步的命令行安装前需要先检查一下自己的驱动版本