先查看显卡和激活环境
nvidia-smi
conda activate pytorch_1.13
然后使用下述代码测试
import torch
print(torch.__version__)
print(torch.cuda.is_available())
x = torch.randn(4,2)
print(x)
先查看显卡和激活环境
nvidia-smi
conda activate pytorch_1.13
然后使用下述代码测试
import torch
print(torch.__version__)
print(torch.cuda.is_available())
x = torch.randn(4,2)
print(x)