我的cuda版本为11.2.1
在pytorch官网上没有找到对用的版本
就安装了自己windows上cuda=10.2的版本试试
torch.cuda_activate()可以通过
pytorch=1.10.1
# CUDA 10.2
conda install pytorch==1.10.1 torchvision==0.11.2 cudatoolkit=10.2 -c pytorch
我的cuda版本为11.2.1
在pytorch官网上没有找到对用的版本
就安装了自己windows上cuda=10.2的版本试试
torch.cuda_activate()可以通过
pytorch=1.10.1
# CUDA 10.2
conda install pytorch==1.10.1 torchvision==0.11.2 cudatoolkit=10.2 -c pytorch