在跑另外一篇文章的代码时,报了和上篇文章不同的错误:
cudatoolkit和torch在这个虚拟环境里都有安装,且这个是成功的
解决方法:若代码中有这一段代码:
这表示环境变量会用哪一块GPU,但服务器上只有一块GPU,其编号是0.(将2改为0)
nvidia-smi
在跑另外一篇文章的代码时,报了和上篇文章不同的错误:
cudatoolkit和torch在这个虚拟环境里都有安装,且这个是成功的
解决方法:若代码中有这一段代码:
这表示环境变量会用哪一块GPU,但服务器上只有一块GPU,其编号是0.(将2改为0)
nvidia-smi