安装Anaconda不要使用sudo,最后一项勾选yes
先安装Pytorch,注意Pytorch有GPU版本和CPU版本,在windows上安装需要指定--index-url
也可以直接安装lightning,会自动安装torch
安装CUDA需要与Torch版本适配,torch.version.cuda # 查看当前torch所对应的cuda版本
推荐使用runfile方式安装cuda,记得在.bashrc中添加环境变量
推荐使用tarball方式安装cuDNN,将
将lib文件移动到/usr/local/cuda/lib64
将include中文件移动到/usr/local/cuda/include