本人使用的显卡如下,打开任务管理器查看
Anaconda下载哪个版本都可以
使用命令conda create -n pytorch python=3.6创建一个名为pytorch的环境,解释器使用3.6的
使用命令conda activate pytorch进入该环境
进入pytorch官网,选择下列选项
复制上面的命令,在pytorch环境下载
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
进入python,导入 torch 包
在pytorch下使用 conda list 命令
使用命令conda uninstall pytorch卸载 cpu 版本的
选择 pip 方式下载,会下载Gpu版本
复制命令到pytorch环境下载
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
没有找到匹配的版本,所以是11.7的版本太高,修改为 11.3版本,继续下载
最后检验,安装成功