(1)按照官网说明即可,非常简单,不同的环境,都有选项:
https://pytorch.org/get-started/locally/
其中查询cuda版本命令:
查看 CUDA 版本:
cat /usr/local/cuda/version.txt
(1)按照官网说明即可,非常简单,不同的环境,都有选项:
https://pytorch.org/get-started/locally/
其中查询cuda版本命令:
查看 CUDA 版本:
cat /usr/local/cuda/version.txt