pytorch安装
1.Anaconda安装
下载链接
直接安装,能选择添加系统路径就添加。
不行的话,手动添加系统环境变量:
2.创建环境
启动Anaconda Prompt 创建虚拟环境:
创建名为 pytorch
的环境,代码:
conda create -n pytorch python=3.8
创建完成后,查看是否成功:
conda info --envs
3.安装pytorch
激活 pytorch
环境:
conda activate pytorch
官网选择对应版本的安装代码,没有对应版本则去查找往期版本的:
conda install pytorch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 pytorch-cuda=11.6 -c pytorch -c nvidia
注意cuda版本需要与电脑显卡匹配。可用nvidia-smi
查看电脑cuda版本。
为了加速可以更改镜像下载路径,更改C:\Users\用户名
路径下的.condarc
文件:
channels:
- defaults
show_channel_urls: true
default_channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
安装完成即可。
测试是否成功:(在pytorch环境内)
python
import torch
torch.cuda.is_available