cmd 中输入:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
cmd 中输入:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda create -n 新环境名 python = 版本号(新环境名自己随便取)
这里我用的:conda create -n pytorch1.4 python=3.7
conda activate 环境名
这里我的是:conda activate pytorch1.4
到 pytorch 官网根据自己的情况进行选择:https://pytorch.org/
其中要选 conda, CUDA 版本的选择会要求电脑英伟达显卡驱动版本必须大于某个值,具体自行查。
将最下面的一行复制下来(末尾的 -c pytorch 不要复制,否则下载的源会默认是 pytorch 官方源,速度贼慢),我的是:
conda install pytorch torchvision cudatoolkit=10.1
在(pytorch1.4) C:\Users\29530>
后输入:python
回车 进入 python 环境,而后再输入import torch
回车无异常表示 pytorch 安装成功, 然后输入: torch.cuda.is_available()
,若返回 true
, 表示该 pytorch 可以使用 GPU。
输入:conda deactivate
删除环境(不要乱删, 这里只是说怎么做):
conda remove -n 环境名
删除清华源并切换回原版源(不要乱删, 这里只是说怎么做):
conda config --remove-key channels
https://blog.csdn.net/qq_42902264/article/details/105299796