目录
1.安装CUDA
首先,在命令行输入nvcc -V检验是否已安装CUDA
若结果显示为
则表明已经配置过CUDA。
若结果为
则尚未配置,配置过程可见笔者的另一篇文章CUDA安装教程(超详细Windows版)
2.安装pytorch
1、检查自己电脑的gpu与cpu
在左下角搜索栏中输入设备管理器,从中找到显示适配器即可看到电脑里的gpu与cpu了。
这里就可以看出你是否有gpu了。
2.查询你的python版本
ctrl+r后输入python
3.安装pytorch
1、进入官方页面pytorch
找到这个页面
从中根据我们的 版本在Run this Command中找到其所对应的网址
https://download.pytorch.org/whl/cu117从中找到我们所要安装的torch torchvision torchaudio
会相对复杂一点。
2、pytorch的whl下载地址
直接进入这个页面开始查找
这里cpu是安装cpu版本的cu是gpu版本的,cp指的是你的python版本,还有就是win和linux系统的选择,找到你所对应的版本将其点击下载。
找到这三个文件将其地址复制到你python的配置环境位置。
在这里进入命令窗口或者cd到这里
对这三个文件分别pip安装,
要按照torch,torchvision,torchaudio的顺序安装完成。
3.检验pytorch安装是否完成
如果结果为True则安装成功。