windows10安装pytorch

1.下载Anconda
2.构建虚拟环境conda create -n 你需要的名字
3.修改镜像源(清华或者上海交大)
4.https://pytorch.org/(安装对应版本的pytorch)
(去掉-c pytorch 表示不去pytorch官网下载,到设置的镜像源去下载)注意python的版本!!!
5.添加到pycharm中
6.如果要安装tensorboard或者tensorboardx,需要先安装tensorflow
(注意tensorflow版本和python版本的对应关系)

conda create -n pytorch_py3.6 python=3.6
conda activate pytorch_py3.6
conda deactivate

nvidia-smi 查看cuda 对应的版本


使用清华镜像
 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
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/


测试pytorch
import torch
torch.cuda.is_available()
torch.cuda.get_device_name(0)

不同的Python对应不同的TensorFlow(注意!!!)

安装tensorflow
pip install tensorflow-gpu==1.12
conda install tensorboard
conda install tensorboardx

pytorch的tensorboard是基于tensorflow的,因此要想使用tensorboard需要安装tensorflow
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值