本地cuda版本、pytorch安装

刚有个4060的本，但是找半天不知道怎么搭环境，折腾了好久，现在记录一下全过程

一、安装anaconda

直接安装即可

http://www.anaconda.com/

判断是否安装成功：打开anaconda prompt即可

二、搭建虚拟环境

在anaconda prompt中输入

conda create -n lzp38 python=3.8

即可安装成功

conda activate lzp #用于激活该环境
conda deactivate #退出该环境
pip list #查看当前环境下有哪些工具包
conda env list #查看当前有哪些环境
conda remove -n lzp --all #删除环境、卸载环境

三、安装torchvision（真正的pytorch）

主要借鉴这一篇
tensorflow、paddlepaddle、pytorch安装

1 先装好cuda和对应的cudnn

对应关系在这里：CUDA Toolkit和cuDNN版本对应关系（更新至2022年6月，附官网永久更新链接）

我的版本号是：

conda install cudatoolkit=11.7.0
conda install cudnn=8.4

2 去官网找对应cuda的torch版本

官网

conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia

四、验证即可

conda activate lzp38
python
import torch
print(torch.version.cuda)
print(torch.cuda.is_available())
print(torch.cuda.get_device_name())
quit()