Pytorch-GPU安装记录(已安装Anaconda和Pycharm)

Pytorch-GPU安装记录(已安装Anaconda和Pycharm)

配置与对应版本

· 系统:Win10
· Anaconda: Anaconda3–4.5.13
· CUDA版本:CUDA 9.0
· cudnn版本:cudnn-9.0-windows10-x64-v7.4.1.5
· Pytorch版本:Pytorch-gpu 1.1.0
· Pycharm版本:JetBrains PyCharm Community Edition 2018.1.2 x64

关于PyTorch及Anaconda的介绍,推荐一个博文,写的挺好~

1. 首先需要查看CUDA对应的版本,查看方式:

open CMD -> nvcc --version

在这里插入图片描述
2. 打开Anaconda, 创建虚拟环境,推荐一个创建环境的博文
Anaconda创建环境的界面
点击Creat新建环境
在这里插入图片描述
官方文档:
https://conda.io/projects/conda/en/latest/user-guide/tasks/manage-environments.html

3. 安装对应的PyTorch版本
Anaconda创建环境界面打开Terminal:
在这里插入图片描述
查看对应CUDA的PyTorch版本

在这里插入图片描述
在打开的Terminal中输入对应代码:

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

然后耐心等待…安装完毕后在终端输入:

import torch
torch.__version__

查看下载版本,如不报错,即安装完毕

4.Pycharm中使用前面创建的虚拟环境,查看及更换环境见此博文~

总的来说,安装PyTorch-GPU比安装Tensorflow-GPU简单多了,后面就可以在新的环境中各种安包了
PyTorch和Tensorflow分别使用了两个环境,但是安装在了一个F盘的大环境下,后面再尝试分开吧~

-------手动分割线-------

记录一个安装完torch后在pycharm中使用时且令人头秃的报错:

Please note and check the following:

  * The Python version is: Python3.7 from "D:\anaconda\envs\qr_env\python.exe"
  * The NumPy version is: "1.19.1"

and make sure that they are the versions you expect.
Please carefully study the documentation linked above for further help.

Original error was: DLL load failed: 找不到指定的模块。

这是由环境变量错误引起的,即Anaconda中可以 import torch, import numpy 啥的,但是在pycharm中环境变量没有添加进去而导致的错误,这里推荐一个靠谱的博文:https://blog.csdn.net/qq_40859802/article/details/107786249

我的解决路径:

1. 查看Anaconda安装路径,由于安装过于久远,所以查找起来费了些时间,方法如下:

Anaconda Prompt中输入 conda info --env 查看路径 ->
6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjYxMzY2NQ==,size_16,color_FFFFFF,t_70)
2. 然后点 Run -> Edit Configurations -> Environment variables -> 添加自己Anaconda的根路径

图片来自 https://blog.csdn.net/qq_40859802/article/details/107786249
在这里插入图片描述
上述两图来自:https://blog.csdn.net/qq_40859802/article/details/107786249

3. 在添加环境变量时,PATH=${Anaconda的根路径}\Library\bin,即查找的根路径+\Library\bin,本文是:PATH = F:\IPYTHON\Library\bin

大功告成!网上很多人让重装,卸载包啥的。有时候卸着卸着人就崩溃了,所以还是多看些解决办法自己想想哪条路可能性更大,然后再去尝试吧~

再推荐一个博文:
https://blog.csdn.net/qq_28714865/article/details/99111240?utm_medium=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromBaidu-1.control&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromBaidu-1.control

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值