标签：1.7 torch print pytorch cuda win10 gpu 252522%

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

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。