Pytorch安装
1.Anaconda的下载和安装
在官网进行下载,一路next(可以修改安装路径,默认安装在C盘),安装完成后可以在菜单看到新增了一些文件
打开这个Anaconda Prompt应用进入之后是以(base)开头,即说明安装成功。
注:如果想要卸载重新安装,先在安装路径中找到Uninstall-Anaconda3.exe
,双击执行进行卸载;然后一定要删除这两个文件:C:\Users\用户名\ .condarc
与C:\Users\用户名\ .conda
,后面的两个文件一定要删除,否则重新下载安装会缺少大量文件(亲自尝试多次才找到原因),然后再重新安装就没问题了。
2.创建激活pytorch环境
根据安装的python版本安装pytorch版本,win+R,打开cmd控制台,输入python就可以看到python版本。
以3.8为例:打开Anaconda Prompt应用输入以下命令创建一个虚拟环境
conda create -n pytorch python=3.8
弹出提示后,输入y开始安装,安装成功后输入以下指令查看已有的虚拟环境
conda info --envs
输入以下指令激活环境(进入环境),会发现路径前由base变成了pytorch,说明成功进入pytorch环境
conda activate pytorch
3.安装验证pytorch
在官网根据自己的情况选择,下面会自动生成运行指令(我这里没有显卡使用的是CPU),在上述pytorch环境中运行生成的指令,即可安装完成。
(在cmd窗口输入命令nvidia-smi
查看自己是否有显卡以及显卡版本号)
conda install pytorch torchvision torchaudio cpuonly -c pytorch
在命令行左边为pytorch环境中,输入python
;然后输入import torch
,如果没有报错意味着Pytorch安装成功;再输入torch.cuda.is_available()
,如果显示True意味着可以使用GPU。