在Ubuntu上獲取PyTorch資源,可參考以下方法:
conda create -n pytorch_env python3.9,激活后conda install pytorch torchvision torchaudio pytorch-cuda11.8 -c pytorch -c nvidia(將11.8替換為所需CUDA版本)。pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118,指定版本安裝則修改對應版本號。nvidia - smi查看顯卡及CUDA版本,再從NVIDIA官網下載對應版本的安裝包進行安裝。