目录
前言:
如果安装了支持CUDA版本的pytorch,就可以启用显卡运算了。
torch.cuda用于设置和运行CUDA操作,会记录当前选择的GPU,并且分配的所有CUDA张量将默认在上面创建,
可以使用torch.cuda.device上下文管理器更改所选设备。
方法一:命令框
Windows+R→cmd→输入:nvcc --version
C:\Users\Administrator>nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:54:10_Pacific_Daylight_Time_2020
Cuda c