首页 > 其他分享 > Pytorch环境安装

Pytorch环境安装

时间:2023-02-11 21:25:25浏览次数:49  
标签:win torch 环境 pytorch Pytorch conda CUDA 安装

WIN10、NVIDIA GeForce RTX 3060

python 3.7,CUDAv11.1.1,PyTorch 1.9.0


1.安装anacodah和PyCharm:

  1.1为了稳定,此处安装了2019年10月16日的Anaconda3-2019.10-Windows-x86_64.exe

  1.2 更换清华源:(更换 conda 源,将默认的国外源更换成国内源,显著提升相关库的下载速度。)编辑用户目录下的 .condarc 文件即可更换 conda 默认源。

  Windows 用户无法直接创建名为 .condarc 的文件,需要先执行如下命令,生成该文件后再修改。

conda config --set show_channel_urls yes

在 .condarc 文件中添加清华源:

ssl_verify: true
show_channel_urls: true
channels:
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/

  1.3安装PyCharm学生邮箱注册账号免费使用PyCharm专业版,无学生邮箱可以下载Community社区版使用。

 


2.创建虚拟环境:

#创建新的虚拟环境,环境名为pytorch,python版本为3.9,都可以根据需求来指定
conda create -n pytorch python==3.9
#删除虚拟环境
conda remove -n pytorch --all
#切换虚拟环境
conda activate pytorch

#查看当前所有虚拟环境,*表示当前所在环境
conda env list


3.安装CUDA:

   不建议直接安装最高版本的CUDA,因为如果后续使用PyTorch或TensorFlow,大概率不兼容。笔者安装的为CUDA Toolkit 11.1.0

#CUDA版本查询
nvcc -V


 

4.配置cuDNN:

  30系显卡最低支持cuDNN v8.0.5,把解压的文件复制,粘贴到CUDA的安装根目录。

  在cmd下进入到安装目录中C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\demo_suite,然后直接执行bandwidthTest.exe和deviceQuery.exe程序,如果返回结果都是pass,那么恭喜,安装成功!


5.安装pytorch:

 首先安装cudatoolkit:

conda install cudatoolkit

  进入pytorch官网,选择想要安装的版本,可以点击REVIOUS VERSIONS OF PYTORCH选择安装历史版本,笔者选择安装CUDAv11.1。

  网络好的直接在线安装:

# CUDA 11.1
pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html

  网络不好的下载torch-1.9.0+cu111-cp37-cp37m-win_amd64.whltorchvision-0.10.0+cu111-cp37-cp37m-win_amd64torchaudio-0.9.0-cp37-cp37m-win_amd64.whl离线安装:

#不加 --no-deps 报错:
#ERROR: Could not find a version that satisfies the requirement typing-extensions (from torch) (from versions: none)
#ERROR: No matching distribution found for typing-extensions
pip install --no-deps torch-1.9.0+cu111-cp37-cp37m-win_amd64.whl
pip install --no-deps torchvision-0.10.0+cu111-cp37-cp37m-win_amd64.whl
pip install --no-deps torchaudio-0.9.0-cp37-cp37m-win_amd64.whl
## 按顺序安装,不然可能安装torchvision的时候给你安装了cpu版本的torch
#卸载
conda uninstall pytorch
python
>>> import torch

  报错:UserWarning: Failed to initialize NumPy

pip install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple

  继续

>>> torch.__version__
>>> torch.cuda.is_available() ## 一定要输出True才是成功


参考:

 第五步:RTX 3060配置CUDA和cuDNN、安装PyTorch

PyTorch环境搭建

笔记本电脑深度学习服务器搭建丨RTX3060+win11+cuda11.0+cudnn+torch

第五步:RTX 3060配置CUDA和cuDNN、安装PyTorch

 

标签:win,torch,环境,pytorch,Pytorch,conda,CUDA,安装
From: https://www.cnblogs.com/Liuyaaa/p/17111775.html

相关文章

  • LAMP环境搭建——最详细的手工编译
    环境:阿里云服务器ECS,AlibabaCloudLinux3.2104LTS64位,2核(vCPU)2GiBLAMP是搭建Web应用时最常用的环境,LAMP分别表示Linux、Apache、MySQL以及PHP,即在Linux......
  • 7.1 运行环境=操作系统+硬件
    由于同一个程序能被大量用户使用,所以说程序具有很大的价值。但是,如果运行环境不同,程序是无法运行的。如表7-1所示:在表示程序的运行环境时,列出了OperatingSystem(操作系统......
  • Linux 环境变量配置的 6 种方法,建议收藏!
    Linux环境变量配置在自定义安装软件的时候,经常需要配置环境变量,下面列举出各种对环境变量的配置方法。下面所有例子的环境说明如下:系统:Ubuntu14.0用户名:uusama需要配......
  • 虚拟机系统安装过程
    1.虚拟机启动后会出现下面的界面(白色表示选中)默认选中的是Testthismedia&installCentOS 7。    InstallCentOS7                    ......
  • Grafana 系列文章(十四):Helm 安装Loki
    前言写或者翻译这么多篇Loki相关的文章了,发现还没写怎么安装......
  • git - 安装地址,基本命令,设置配置,
    1.Git安装地址:https://git-scm.com/downloads 2.基本命令#查看配置gitconfig-l#查看Git所有配置gitconfig--system--list#查看系统配置gitconfig--global......
  • JDK的下载及安装
    下载地址:https://www.oracle.com/java/technologies/downloads/.  选择系统下载不同格式的安装包。PS:事实是zip的下载很快安装以JDK8为例:点击exe文件安装,按照提示安......
  • Linux上安装jdk Tomcat mysql redis等
    1.安装JDk  1.1这里使用xshell中xfxp进行文件的上传,将jdk二进制包上传到Linux服务器上 下载地址:https://www.oracle.com/java/technologies/downloads/#java8  ......
  • thttpd 2.27(最新)移植指南(官方安装脚本好多坑,我只想说)
    PS:要转载请注明出处,本人版权所有。PS:这个只是基于《我自己》的理解,如果和你的原则及想法相冲突,请谅解,勿喷。前置说明  本文作为本人csdnblog的主站的备份。(BlogID......
  • PyQt5 Ubuntu 16.04/14.04 环境配置
    PS:要转载请注明出处,本人版权所有。PS:这个只是基于《我自己》的理解,如果和你的原则及想法相冲突,请谅解,勿喷。前置说明  本文作为本人csdnblog的主站的备份。(BlogID......