# device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") os.environ["CUDA_VISIBLE_DEVICES"]="0,1" # test=torch.cuda.is_available() print(torch.cuda.device_count()) print(torch.cuda.get_device_name(1)) print(torch.cuda.current_device()) torch.cuda.set_device(0)https://developer.nvidia.com/cuda-toolkit-archivecuda 版本列表 显卡算力查看命令 nvidia-smicuda下载
cudnn 下载

1525

被折叠的 条评论
为什么被折叠?



