(1)按照官網(wǎng)說明即可,非常簡單,不同的環(huán)境,都有選項(xiàng):
? ? ? ? ? ?https://pytorch.org/get-started/locally/

其中查詢cuda版本命令:
查看 CUDA 版本:
cat /usr/local/cuda/version.txt
(1)按照官網(wǎng)說明即可,非常簡單,不同的環(huán)境,都有選項(xiàng):
? ? ? ? ? ?https://pytorch.org/get-started/locally/
其中查詢cuda版本命令:
查看 CUDA 版本:
cat /usr/local/cuda/version.txt