安装torch
1. 安装好pycharm,anaconda
2. 再win+r,运行cmd,输入nvidia-smi后回车
3. 表中的Driver Version需大于396.26,否则就去geforce experience更新显卡吧。
4. 桌面搜索栏搜索打开
anaconda prompt
5. 更改清华镜像源下载会更快
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/free/
6. 创建虚拟环境:conda create -n your_env_name
7. 打开虚拟环境:conda activate your_env_name
8. 下载torch:https://pytorch.org/官网
9. 有GPU选10的,无选9
10. 复制Run this Command的代码(若改了镜像源则删掉 -c pytorch),粘贴进虚拟环境中
11. 安装好后输入python进入python环境,输入import torch,若无报错证明已安装成功。
12. 备注:以后要在anaconda安装其他软件包需要进入虚拟环境中安装
a. 进入anaconda prompt
b. 打开虚拟环境:conda activate your_env_name
c. 安装所需软件包
另:使用python开发工具时要改一下interpreter
例如:在pycharm中file新建项目new project,在existing interpreter选python xx(your_env_name),然后create。
完整教程(别人的):https://www.jb51.net/article/186132.htm
另一种方法:
conda create -n pytorchh python==3.8
conda activate pytorchh
conda install pytorch torchvision torchaudio cudatoolkit=10.2
torch.cuda.is_available()
安装完后在pycharm重建文件(编译器要改成下载了torch的)