windows下安装pytorch
前言:
最近换了一台笔记本电脑(3060显卡),单单为了配这个pytorch环境就花了1天时间(即使我以前配过)。所以,现在想要记录一些细节防止忘记。
1.anaconda事先安装好。
2.anaconda默认是不走代理的,导致了我们下载pytorch的话就不能挂梯子了。(借助conda下载
一挂梯子就会报proxy等错误,就很烦。我们只能从官网下载,而官网下载又巨慢(有时也可行,可能是看时间点)。
最后,从清华的镜像源下载。
3.30系显卡的cuda要求11以上,别装错了。装错了程序就跑不起来的。
*知道anaconda怎么走代理的大佬,欢迎在评论区留言(麻烦具体一点点
*注意:pip可以走代理,也就是说当我们在pip install somepackages
的时候可以挂梯子的。我以前一直分不清,现在懂了。
*本来安装是很简单的,就是一条命令。tmd都是因为墙,搞来搞去,才这样麻烦。
配置
1.conda创建虚拟环境(指定python版本
单击Anaconda Prompt进入终端来创建虚拟环境。
conda create -n pytorch python=3.7
2.进入虚拟环境
conda activate pytorch
3.给conda配置清华镜像(因为下载是借助conda来下的
相当于是说,我换一个网站下载,东西一样的然后速度快一亿点。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
4.下载pytorch
https://pytorch.org/get-started/locally/ <–pytorch下载地址
conda install pytorch torchvision torchaudio cudatoolkit=11.3
*注意:把-c pytorch
去掉,不然的还是从官网下载。(-c pytorch
的意思是 pytorch下载的channel是官网
4.测试
输入
python
import torch
torch.cuda.is_available()
是True就好了。
=================================================================================================
conda其他命令
退出当前虚拟环境
conda deactivate pytorch
环境里面装了什么
conda list
删除虚拟环境
conda remove -n pytorch --all
看看我有多少虚拟环境
conda-env list