conda create -n py36+torch python=3.62. 换源
生成.condarc文件
conda config
修改文件内容:
channels: - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ - defaults3. 安装pytorch
进入 pytorch官网
选择对应的版本:
开始安装:
conda install pytorch torchvision torchaudio cpuonly -c pytorch4. 验证
conda activate py36+torch python import torch
如果安装GPU版本,首先确保安装了对应的NVIDIA显卡驱动
然后 print(torch.cuda.is_available())
结果为True,说明GPU可用



