win10,创建新环境并安装pytorch-gpu=1.7.0版本

之前装过gpu版本tensorflow,包含cudatoolkit=10.1 ,创建新环境装gpu版本pytorch时考虑是否再装cudatoolkit?以下是没有再装cudatoolkit情况下,目前正常。

文章目录

  • Anaconda 创建新环境
  • 一、安装gpu版本pytorch
  • 二、测试安装是否成功?
    • 1.测试代码
    • 2.输出结果


Anaconda 创建新环境

anaconda prompt下命令如下(环境名称我命名为pytorch):

conda create --name pytorch python==3.7.5

期间按提示输入 y 即可。完成后激活新环境,代码如下:

activate pytorch

已激活,现在是在新环境下:在这里插入图片描述

一、安装gpu版本pytorch

命令行(包括torch、torchvision等版本):

pip install torch==1.7.0+cu101 torchvision==0.8.1+cu101 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple

安装过程曾出现警告:
只需再环境变量里添加路径即可,参考:https://blog.csdn.net/weixin_44725869/article/details/107380918?ops_request_misc=%25257B%252522request%25255Fid%252522%25253A%252522161095737116780271569822%252522%25252C%252522scm%252522%25253A%25252220140713.130102334.pc%25255Fall.%252522%25257D&request_id=161095737116780271569822&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2allfirst_rank_v2~rank_v29-5-107380918.pc_search_result_hbase_insert&utm_term=%20WARNING:%20The%20script%20f2py.exe%20is%20installed%20in%20%27D:\Miniconda3\envs\Scripts%27%20which%20is%20not%20on%20PATH.
win10,创建新环境并安装pytorch-gpu=1.7.0版本_第1张图片

我已安装,下图仅供展示:win10,创建新环境并安装pytorch-gpu=1.7.0版本_第2张图片

二、测试安装是否成功?

1.测试代码

代码如下(示例):

#coding:utf-8
import torch
print(torch.cuda.current_device())# 输出当前设备(我只有一个GPU为0print(torch.cuda.device(0))
print(torch.cuda.device_count())# 输出含有的GPU数目
print(torch.cuda.get_device_name(0))# 输出GPU名称 --比如GeForce RTX 2060
x = torch.rand(5, 3)
print(x)

2.输出结果

若正常输出下图内容即可:
win10,创建新环境并安装pytorch-gpu=1.7.0版本_第3张图片

你可能感兴趣的:(python,pytorch)