在Ubuntu上解決PyTorch沖突可按以下步驟操作:
conda或venv創建虛擬環境,避免系統級包沖突。conda create -n pytorch_env python=3.9 # 指定Python版本
conda activate pytorch_env
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
# 或使用pip
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
pip/setuptools,或手動指定沖突包版本。pip install --upgrade --force-reinstall 包名
import torch
print(f"PyTorch版本: {torch.__version__}")
print(f"CUDA可用: {torch.cuda.is_available()}")
nvidia-smi查看。若仍無法解決,建議查閱PyTorch官方文檔或社區論壇,提供具體錯誤信息進一步排查。