如何用conda安装PyTorch(windows、GPU)最全安装教程(cudatoolkit、python、PyTorch、Anaconda版本对应问题)(完美解决安装CPU而不是GPU的问题)
在Windows上使用conda安装PyTorch及其依赖(包括CUDA toolkit和Python),您可以遵循以下步骤:
- 打开Anaconda Prompt。
(可选)创建一个新的conda环境:
conda create -n myenv python=3.8 conda activate myenv
替换
myenv
为您希望命名的环境名称,并设置所需的Python版本。安装PyTorch和CUDA toolkit。您可以直接访问PyTorch官网的安装指南页面(https://pytorch.org/get-started/locally/)来获取最新的安装命令。它会根据您的系统配置(如CUDA版本)生成适当的命令。
例如,如果您有支持CUDA的NVIDIA GPU,并希望安装与CUDA 11.3兼容的PyTorch版本(请确保下载与您的CUDA版本匹配的wheel),您可以使用以下命令:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
请注意,CUDA版本(例如11.3)需要与您的NVIDIA驱动兼容。您可以通过
nvidia-smi
命令查看安装在您系统上的CUDA版本。
以上步骤会安装PyTorch及其相关依赖,包括CUDA toolkit,在您的conda环境中。如果您没有GPU或不想使用GPU,可以省略cudatoolkit
的安装,只需要安装CPU版本的PyTorch:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
请确保访问PyTorch官网以获取最新的安装命令和可用版本。
评论已关闭