Windows系统安装PyTorch
PyTorch官网点击这里:PyTorch官网
创建虚拟环境
打开Anaconda Prompt,创建虚拟环境,如创建一个名字为pytorch的运行环境
conda create -n pytorch python=3.9
查看环境
conda info --envs
切换环境
conda activate pytorch
删除环境
conda remove -n pytorch --all
CUDA版(需要NVIDIA驱动程序)
检查CUDA Driver版本
打开Anaconda Prompt,输入
nvidia-smi
加载已有whl安装包
pip install "torch-1.7.1+cu101-cp39-cp39-win_amd64.whl" -i [https://pypi.tuna.tsinghua.edu.cn/simple](https://pypi.tuna.tsinghua.edu.cn/simple)
pip install "torchvision-0.8.2+cu101-cp39-cp39-win_amd64.whl" -i [https://pypi.