1.环境配置
安装anaconda: 直接在anaconda.com官网里面下载然后安装
查看显卡信息和对应的可使用的最高等级的cuda版本:nvidia-smi
然后通过conda创建env:conda create -n envpytorch python=3.12.07
激活虚拟环境: conda activate envpytorch
安装pytorch(Cuda会跟着一起安装下来): 直接在pytorch.com官网的指令在对应的虚拟环境里面安装
测试在env里是否安装成功pytorch: python; import torch; torch.cuda.is_available()-->True;
在vscode右下角可以选择当前文件使用的解释器,选择env\envpytorch解释器就行
2.使用tips
dir()可以查看某个包里面含有什么东西,如dir(torch)
help()解释某个函数的用法,如help(torch.cuda.is_available)