在Linux系統更新PyTorch版本,常用方法有使用pip和conda兩種,具體如下:
pip uninstall torch torchvision torchaudio。nvcc --version或nvidia - smi。pip install torch torchvision torchaudio --extra - index - url https://download.pytorch.org/whl/cu117;若用CPU版本,去掉--extra - index - url參數即可。conda activate your_environment_name。conda remove pytorch torchvision torchaudio -c pytorch。conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch;CPU版本則用conda install pytorch torchvision torchaudio cpuonly -c pytorch。更新前建議備份項目代碼和數據,確保CUDA和cuDNN版本與新版本PyTorch兼容。