win11下pycharm 和anaconda配置pytorch（CPU+GPU）

win11下pycharm 和anaconda配置pytorch（CPU+GPU）

我的电脑：win11+RTX3060
1、配置CPU版的pytorch
直接去torch官网，选择CPU即可

直接在终端运行这句代码，即可配置完成。

conda install pytorch torchvision torchaudio cpuonly -c pytorch

我的pycharm解释器是anaconda的，所以直接是装在了这个环境下。

2、配置GPU版的torch
先看自己电脑支持的cuda版本，我的是11.3
在英伟达控制面板里查看。

这里显示我支持11.3 的cuda

选择cuda 11.3

终端运行代码

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

3、测试

测试代码：

import torch
#CPU,GPU
x = torch.rand(4, 6)
print(x)
#GPU
print(torch.cuda.is_available())