Pytorch – .to()和.cuda()的区别
1 Pytorch中.to()和.cuda()的区别 如果需要指定的设备是GPU则.to()和.cuda()没有区别，如果设备是cpu，则不能使用.cuda()。也就是说.to()既可以指定CPU也可以指定GPU，而.cuda()只能指定GPU。 1.1 .cuda() 1.单GPU os.envi…
- Pytorch
- 2022-07-11
CUDA 安装报错 could not create file “…\chrome_elf.dll”
1 CUDA 安装报错 could not create file "...\chrome_elf.dll" 今天重新在win 7上安装CUDA 10.2的时候报了这个错，人都懵了，我以为是卸载之前的CUDA 10.0没有卸载干净导致注册表有残留信息，然后重启了电脑再安装，然后在解压exe的时候依然…
- 人工智能
- 2021-09-10
Pytorch – 检测CUDA、cuDNN以及GPU版本的Pytorch是否安装成功、GPU显存测试
1 测试代码 import time import torch if __name__ == '__main__': print('torch版本：'+torch.__version__) print('cuda是否可用：'+str(torch.cuda.is_available())) print…
- Pytorch
- 2021-08-06
Python – Linux/Centos/Ubuntu查看CUDA/cuDNN版本号
1 查看CUDA版本号 低版本使用： nvcc -v 高版本使用： nvcc --version 查看CUDA版本号。 2 查看cuDNN版本号 旧版本cuDNN使用以下命令查看版本号： cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -…
- Python
- 2021-06-11