准备工作
硬件:
准备一张性能怪兽级的NVIDIA显卡3060TI
查询链接:link.
软件安装
本文使用软件网盘下载链接:
链接:https://pan.baidu.com/s/1rgqW0mFJ-9gLgerPqzNmsg
提取码:wcjl
1.安装NVIDIA显卡驱动>511.65
这里安装3060TI的522.30Studio版驱动
522.30-desktop-win10-win11-64bit-international-nsd-dch-whql.exe
2.安装cuda_11.6
cuda_11.6.2_511.65_windows.exe
3.安装cudnn
解压cudnn-windows-x86_64-8.4.1.50_cuda11.6-archive.zip压缩包里面的文件,放到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6目录下
4.安装Miniconda3
Miniconda3具有简洁易用的特点,推荐使用
Miniconda3-py39_4.10.3-Windows-x86_64.exe
5.安装pytorch1.13版本
官网链接:https://pytorch.org(其他版本看官方提供的安装要求)
打开Anaconda Prompt (miniconda3),激活安装虚拟环境
输入conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia回车
6.ok搞定,开始写代码!!!