python安装pytorch-cpu(Window系统64bit)

【时间】2019.08.15

【题目】python安装pytorch-cpu(Window系统64bit)

参考官网:https://pytorch.org/get-started/locally/

需要安装torch和torchvision库

一、使用anaconda安装:

conda install pytorch-cpu torchvision-cpu -c pytorch

下载太慢的话直接下载whl文件再安装:http://download.pytorch.org/whl/cpu/torch-1.1.0-cp36-cp36m-win_amd64.whl

修改torch-1.1.0-cp36-cp36可以直接获取对应版本。

本人使用的是python3.6.5,目前的最新版本是torch-1.1.0-cp36-cp36

然后直接cd到下载目录,再安装:

conda install --use-local xxx.whl

但本人安装时出错,最后只能在anaconda prompt中使用pip安装:

pip install xxx.whl

二、使用pip安装

# Python 2.7
pip install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp27-cp27mu-linux_x86_64.whl
pip install torchvision

# if the above command does not work, then you have python 2.7 UCS2, use this command
pip install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp27-cp27m-linux_x86_64.whl
# Python 3.5
pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision
# Python 3.6
pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision
# Python 3.7
pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp37-cp37m-win_amd64.whl
pip3 install torchvision

 三、简单测试安装是否成功

  • python
  • import torch
  • print(torch.__version__)

成功会输出版本号。

 

要在Python安装PyTorch的GPU版本,你需要先确保你的显卡支持CUDA。然后,你可以根据你的操作系统Python版本,按照以下步骤进行安装: 1. 安装CUDA和CUDNN: - 访问NVIDIA官网下载并安装适合你系统的CUDA和CUDNN版本。确保选择与你的显卡和操作系统兼容的版本。 - 安装CUDA时,选择自定义安装,并配置好路径。 2. 创建一个虚拟环境(可选): - 打开终端或命令提示符窗口,并执行以下命令创建一个虚拟环境(可选): ``` python -m venv myenv ``` 3. 激活虚拟环境(可选): - 在终端或命令提示符窗口中执行以下命令激活虚拟环境(如果你使用了虚拟环境): - Windows: ``` myenv\Scripts\activate ``` - macOS/Linux: ``` source myenv/bin/activate ``` 4. 安装PyTorch GPU版本: - 打开终端或命令提示符窗口,并执行以下命令来安装PyTorch GPU版本: - 使用conda(推荐): ``` conda install pytorch torchvision torchaudio cudatoolkit=xx.x -c pytorch ``` 其中,`xx.x`是你安装的CUDA版本对应的数字,例如`10.2`。 - 使用pip: ``` pip install torch torchvision torchaudio ``` 5. 验证安装: - 打开Python交互式环境(终端或命令提示符窗口中输入`python`)并执行以下命令来验证PyTorch是否安装成功: ```python import torch print(torch.cuda.is_available()) ``` 如果输出结果为`True`,说明PyTorch的GPU版本安装成功,并且你的显卡支持CUDA。如果输出结果为`False`,则可能是因为你的显卡不支持CUDA或者安装过程中出现了问题。 请注意,安装PyTorch GPU版本可能会比普通版本复杂一些,并且需要满足一些额外的硬件和软件要求。确保你已经详细阅读了PyTorch的官方文档,并按照要求进行安装和配置。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值