下载anaconda
官网下载(慢),https://www.anaconda.com/
或者镜像下载最新的版本(快),点下镜像网站安装包那里的date,才出来最新的,不知道为啥。下载64位的
https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
- 下载完直接安装就好了
- 注意安装的时候选择Just me , 选All Users会让你后面新创建的环境还在C盘(只有base环境在你自定义的位置,大冤种。。。。)
添加环境变量
- 最新版anaconda不知道为啥,安装时没有添加到环境变量的选项,要自己配置一下,加上下面四个就好了
安装pytorch(最关键的就是用pip安装,指定国内源)
- win+r 打开cmd
conda -V
查看环境变量是否成功conda init
初始化condaconda env list
查看当前有的环境conda create -n pytorch python=3.10
创建叫pytorch的环境conda activate pytorch
切换到该环境- 去pytorch官网,选择用pip安装,复制下要安装的包名称,即下图红框所示,因为conda现在不知道怎么回事,用清华源也很慢
pytorch安装
- 官网看下命令,要安装的包
- 用的交大源
- 科大源不继续更新了,不能用,清华源当时好像没试,感觉是可以的,这里用的如下命令,上交源可以
- 有的源里(比如ustc)没有继续更新pytorch包了,所以你用它的源下载不到,他可能把你下载个他源里有的cpu版本的pytorch
- 然后torch.cuda.is_available()就会报错false。
pip3 install torch torchvision torchaudio --index-url https://mirror.sjtu.edu.cn/pytorch-wheels
- 测试是否成功
- 新建终端运行下python
python
- 敲入如下命令 ,返回true则成功
import torch
torch.cuda.is_available()
cuda和cudnn还是要安装的,我这次安装pytroch官网给的命令没有pytorch-cuda=11.6 了,所以要自己安装下
-
注意先根据自己显卡驱动安装对应的pytorch版本,然后根据pytorch版本安装对应的cuda版本,再根据cuda安装对应的cudnn
-
附一个解释cuda、cudnn、驱动之间关系的帖子