1.安装pytorch
pytorch官方选择合适版本
2.检测pytorch是否安装成功
cpu版本:
1. conda activate 虚拟环境名
2. conda list 看有没有pytorch或者torch
3. 输入python
4. 输入 import torch
5. 输入 torch.cuda.is_available()
6. 如果显示False,就说明我们这个PyTorch安装成功了
3.常用conda命令
3.1 生成环境 / 包
conda env export > environment.yml
pip freeze > requirements.txt
3.2 创建 / 激活 / 退出 / 查看所有 虚拟环境
conda create -n your_env_name python==3.9
conda activate your_env_name
conda deactivate your_env_name
conda env list
3.3 删除指定 / 所有虚拟环境
conda env remove --name your_env_name
conda remove –n --all
3.4 添加 / 删除 / 查看 镜像
conda config --add 镜像地址
conda config --remove 镜像地址
conda config --get
3.5 镜像地址
清华:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
北外:
https://mirrors.bfsu.edu.n/anaconda/cloud/pytorch/win-64/
阿里:
http://mirrors.aliyun.com/anaconda/cloud/pytorch/win-64/
南大:
https://mirror.nju.edu.cn/pub/anaconda/cloud/pytorch/win-64/