最简单方式,查找torch与cuda、python匹配的命令
conda search pytorch
然后输入
import torch;print (torch.__version__)
torch.cuda.is_available() cuda是否可用;
torch.cuda.get_device_name(0) 返回gpu名字,设备索引默认从0开始;
最简单方式,查找torch与cuda、python匹配的命令
conda search pytorch
然后输入
import torch;print (torch.__version__)
torch.cuda.is_available() cuda是否可用;
torch.cuda.get_device_name(0) 返回gpu名字,设备索引默认从0开始;