我的电脑:win11+RTX3060
1、配置CPU版的pytorch
直接去torch官网,选择CPU即可
直接在终端运行这句代码,即可配置完成。
conda install pytorch torchvision torchaudio cpuonly -c pytorch
我的pycharm解释器是anaconda的,所以直接是装在了这个环境下。
2、配置GPU版的torch
先看自己电脑支持的cuda版本,我的是11.3
在英伟达控制面板里查看。
我的电脑:win11+RTX3060
1、配置CPU版的pytorch
直接去torch官网,选择CPU即可
直接在终端运行这句代码,即可配置完成。
conda install pytorch torchvision torchaudio cpuonly -c pytorch
我的pycharm解释器是anaconda的,所以直接是装在了这个环境下。
2、配置GPU版的torch
先看自己电脑支持的cuda版本,我的是11.3
在英伟达控制面板里查看。