PyTorch的安装与使用

我的环境:cuda9.0+window10+python3.5+pycharm+anaconda3

1.anaconda安装

conda install pytorch -c pytorch
pip3 install torchvision

2.pip 安装

pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision

3.安装完成测试
建立新的python文件,输入以下代码

from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)

运行后如果能够出现形状类似如下的矩阵,代表你已经安装成功可以正常使用了

tensor([[0.3380, 0.3845, 0.3217],
        [0.8337, 0.9050, 0.2650],
        [0.2979, 0.7141, 0.9069],
        [0.1449, 0.1132, 0.1375],
        [0.4675, 0.3947, 0.1426]])

注:
1.用笔者给的代码需要参照环境版本,如果没有gpu(cuda)或者版本不一致可到pytorch官网安装教程进行查找对应代码更换。
2.下载过程中有可能会出现time out等下载缓慢等超时现象,原因可能是下载的对应资源在国外,可参考以下代码将源地址设置为国内镜像即可加速。

# 添加Anaconda的TUNA镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
# 设置搜索时显示通道地址
conda config --set show_channel_urls yes

3.这是常用的两种方法,还有一种方法为离线安装法,速度可能更快,可参考其他作者博客。

  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值