PyTorch
2019-09-03.
Category & Tags:
Deep Learning,
PyTorch
Install (w/ GPU) # win10 # Install NV GPU driver and compatible CUDA version first, or install using pip together. See PyTorch doc’s selector to find a compatible CUDA version. Then use the cmd given by the selector to install PyTorch: pip source: # pip install pytorch torchvision torchaudio cudatoolkit=11.1 pip binary: # pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html Tip: the torch...whl file is > 3GB, which can be pre-downloaded via IDM etc. ...