【PYTORCH】Pytorch与GPU相关大合集（含测试代码）

2021-12-25 20:34:22 阅读：265 来源： 互联网

标签：查看 torch PYTORCH print Pytorch device cuda 测试代码 GPU

1.查看pytorch是否存在以及pytorch的版本

import torch
print(torch.__version__)

2.查看cuda是否可用

print(torch.cuda.is_available())

3.查看cuda版本

print(torch.version.cuda)

4.查看当前工作的GPU

print(torch.cuda.current_device())

5.查看gpu的数目

print(torch.cuda.device_count())

6.设置工作GPU

print(torch.cuda.current_device())
torch.cuda.set_device(1)
print(torch.cuda.current_device())

也有的方法说下述方式也可以实现切换GPU但本人亲测无效

os.environ["CUDA_VISIBLE_DEVICES"] = '0'

7.查看gpu名字，设备索引默认从0开始：

print(torch.cuda.get_device_name(0))
print(torch.cuda.get_device_name(1))

标签：查看,torch,PYTORCH,print,Pytorch,device,cuda,测试代码,GPU
来源： https://blog.csdn.net/m0_58810879/article/details/122147996

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9