之前装过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-2_allfirst_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.

我已安装，下图仅供展示：

二、测试安装是否成功？

1.测试代码

代码如下（示例）：

#coding:utf-8
import torch
print(torch.cuda.current_device())# 输出当前设备（我只有一个GPU为0）
print(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.输出结果

若正常输出下图内容即可：

标签：1.7,torch,print,pytorch,cuda,win10,gpu,252522%
来源： https://blog.csdn.net/weixin_49747246/article/details/112788212