1、安裝pytorch
(1)查看CUDA版本
nvcc -v
(2)不同版本對應的pytorch安裝方式
(3)linux上 CUDA 9.0 Python 3.5安裝方法
pip install torch torchvision
2、安裝torchaudio
嘗試了很多方法,最後在linux上直接使用pip成功。注意torchaudio目前只支持在linux和mac上安裝
pip install torchaudio
其他方法:
- conda安裝https://anaconda.org/pytorch/torchaudio
conda install -c pytorch torchaudio
- https://ptorch.com/news/100.html