conda 安装 pytorch

安装pytorch

1. 创建python环境

conda create -n py36+torch python=3.6

激活环境

conda activate py36+torch

2. 换源

生成.condarc文件

conda config

修改文件内容：

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
show_channel_urls: true

linux系统的 去掉后面的 win-64/

3. 安装pytorch

进入 pytorch官网

选择对应的版本：

开始安装：

去掉最后的 -c pytorch，否则还是从conda官方源下载。

conda install pytorch torchvision torchaudio cpuonly

4. 验证

python
import torch

如果安装GPU版本，首先确保安装了对应的NVIDIA显卡驱动
然后 print(torch.cuda.is_available())
结果为True，说明GPU可用