已经成功的安装了pytorch-gpu。
>>> import torch
>>> print(torch.cuda.is_available())
True
>>> quit()
但是nvcc时,找不到cuda
输入:nvcc, 显示:
Command 'nvcc' not found, but can be installed with:
sudo apt install nvidia-cuda-toolkit
以下是在pytorch官方论坛上同样问题的解答。
重新安装:
sudo apt install nvidia-cuda-toolkit
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)