第一步:创建环境 (-n +环境名(torch) + python版本(3.10)) conda create -n torch python=3.8 第二步:了解你电脑的cuda 版本,在terminal中输入:nvidia-smi即可,如下图所示,版本为12.3 第三步:去pytorch官网看你的conda适配的torch、torchvision以及 torchaudio版本,这个链接: Previous PyTorch Versions | PyTorc...
如果要使用GPU进行机器学习的训练,那么首先需要支持训练的显卡及驱动即正确安装CUDA、CUDNN,最重要的一点是需要与驱动对应的torch GPU版本,否则大概率使用torch.cuda.is_available()命令检查GPU是否可用时得到False。 检查显卡-显卡驱动CUDA适配版本-下载Anaconda-下载CUDA-检查CUDA是否安装好-下载CuDNN-下载GPU版本的pyto...