安装pytorch环境
登录pytorch官网开始界面,根据自身情况选择相应选项
打开命令行,输入选项下面的命令:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
登录CUDA11.8下载界面,下载相应CUDA
最后检验是否安装成功
登录pytorch官网开始界面,根据自身情况选择相应选项
打开命令行,输入选项下面的命令:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
登录CUDA11.8下载界面,下载相应CUDA
最后检验是否安装成功