Win10 安装pytorch-gpu版本的坑

用官网指令安装torch是cpu版本的。 不知道安装的是cpu还是gpu版本对可以用torch.__version__查看,如果是xx+cpu就是cpu版本。

如果要安装GPU版本,去torch_stable里面找对应版本. 这里我用的是cu113/torch-1.10.2%2Bcu113-cp36-cp36m-win_amd64.whl理解一下参数:cu113表示需要cuda11.3,torch版本是1.10,2,cp36指需要的是python3.6,win系统.

另外如果之前有装过cuda的其他版本,再次安装不会覆盖.所以装了新cuda测试nvcc -V 还是老版本。只需要删除老版本的cuda的环境变量就可以.

检测pytorch是否可用的代码:

import torch
print(torch.__version__)

另1: 2022.4.29 由于要使用paddlepaddle,目前其只支持到11.2,故更换为11.2 ,torch1.10.2+cu113可兼容cuda11.2

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