安装torch

安装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的)

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值