老实手机

文章
7
资源
0
加入时间
2年10月21天

ubuntu 20.04 安装pytorch、cuda

首先第一步如果你没有换 pip 源的话看这里pip3 install torch torchvision torchaudiowget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pinsudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600sudo apt-key