pytorch 1.8 安装 正常启动torch.cuda

先上图pytorch 1.8 安装 正常启动torch.cuda_第1张图片

pytorch 安装,因为我的版本是cuda是10.1直接用的是镜像指令:

pip install torch==1.8.0+cu101 torchvision==0.9.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple

你可能感兴趣的:(pytorch,深度学习,python)