万恶的环境2——安装的torch版本是cpu版本如何改为GPU版本
目录感谢 参考的链接
[错误Torch not compiled with CUDA enabled解决方法附CUDA安装教程及Pytorch安装教程_nm235666的博客-CSDN博客](https://blog.csdn.net/moyong1572/article/details/119438286#:~:text=运行时报出AssertionError%3A Torch not compiled with CUDA enabled错误 ,具体,” if torch. cuda.is_available () else “cpu”) 代码)
1 报错
2 检查
import torch as th
print(th.__version__)
print(th.cuda.is_available())
cuda1 = th.device('cuda')
print(f'device:{cuda1} device.type:{cuda1.type}')
s = th.randn((2,3),device=cuda1)
print(s)
3 结果与问题——torch版本是cpu版本的
4解决
4.1 首先检查自己受否有安装cuda
在命令行中输入
nvcc -V
4.2 到官网去下载对应版本的torch
5 验证效果
再次运行代码 ,发现成功
标签:版本,torch,th,device,GPU,安装,cpu From: https://www.cnblogs.com/zuti666/p/16975994.html