pytorch安装
一、创建虚拟环境
conda create -n pytorch_env python=3.8
二、激活虚拟环境
conda activate pytorch_env
为确保安装速度,可以修改.condar(window10存储路径为C:\Users\username.condarc)文件,具体内容如下
ssl_verify: true
channels:
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- https://anaconda.org
- defaults
show_channel_urls: true
三、安装命令
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
四、检查是否安装成功
import torch
torch.cuda.is_available()