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, 创建虚拟环境,推荐一个创建环境的博文
点击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
查看路径 ->
2. 然后点 Run -> Edit Configurations -> Environment variables -> 添加自己Anaconda的根路径
上述两图来自:https://blog.csdn.net/qq_40859802/article/details/107786249
3. 在添加环境变量时,PATH=${Anaconda的根路径}\Library\bin
,即查找的根路径+\Library\bin,本文是:PATH = F:\IPYTHON\Library\bin
大功告成!网上很多人让重装,卸载包啥的。有时候卸着卸着人就崩溃了,所以还是多看些解决办法自己想想哪条路可能性更大,然后再去尝试吧~