785
文章
451600
访问
分类
暂无分类
标签
暂无标签
pytorch安装
发布于2020-07-22 22:54 阅读(161) 评论(0) 点赞(1) 收藏(5)
design by ht
3行缩进表要键入的代码
1 anaconda
下载：
https://www.anaconda.com/distribution/#download-section
双击安装
选择 All users
勾选 Add Anaconda3 to the system PATH environment variable
测试是否anaconda是否安装成功
win+ R
cmd
conda
复制下面代码输入
######中科大 #设置搜素时显示通道地址
conda config --add channels https:/mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
conda config --akld channels https://mirrors.ustc.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/menpo/
conda config --set show channel_urls yes
######清华
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show channel_urls yes
######pip 清华
pip config set global.index-url https://mirrors.alyun.com/pypi/simple
阿里云 http://mirrors.aliyun.com/pypi/simple
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple
豆瓣(douban) http://pypi.douban.com/simple
清华大学 https://pypi.tuna.tsinghua.edu.cn/simple
中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple
conda 创建虚拟环境
虚拟环境1 python=3.6 pytorch =1.5.0 （GPU）
conda create -n pytorch_1.5_gpu python=3.6
y
若中科大镜像原安装失败，则换清华镜像源重置
先移除中科大
conda config --remove-key channels
conda config --show
在channels 中有镜像源
复制清华镜像源
进入虚拟环境
conda activate pytorch_1.5_gpu
虚拟环境2 python=3.7 pytorch=1.4.0 （CPU）
conda create -n pytorch_1.4_cpu python=3.7
y
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-g22CpfmL-1595321586142)(C:\Users\ht\AppData\Roaming\Typora\typora-user-images\image-20200721114725353.png)]
进入虚拟环境2
conda activate pytorch_1.4_cpu
退出环境
conda deactivate
2.pycharm
下载：
https://www.jetbrains.com/pycharm/
激活：
https://shimo.im/docs/GJTqgHqh6kGYkKKY/read
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-vfZMv4Oe-1595321586145)(C:\Users\ht\AppData\Roaming\Typora\typora-user-images\image-20200721120240780.png)]
3.CUDA安装
下载：
https://developer.nvidia.com/cuda-gpus#collapseOne
安装步骤：
1.检查pytorch版本所支持的cuda版本，下载对应版本的CUDA
2.下载CUDA安装包，下载对应CUDA版本的cudnn安装包
3.安装并验证安装成功
查看cuda 版本
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin
nvcc -V
结果显示
Copyright © 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:32:27_Pacific_Daylight_Time_2019
Cuda compilation tools, release 10.2, V10.2.89
4.cudnn
下载：https://developer.nvidia.com/cudnn
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite
bandwidthTest.exe
结果显示 Result = PASS 即可
deviceQuery.exe
结果显示 Result = PASS 即可
5.pytorch
命名解释：
cu92/torch-1.2.0%2Bcu92-cp37-cp37m-win_amd64.whl
cuda版本或cup pytorch版本号 python版本号 操作系统
5.1 配置虚拟环境1
虚拟环境1 python=3.6 pytorch =1.5.0 （GPU）
https://download.pytorch.org/whl/torch_stable.html
Ctrl +F 搜索 cu102/torch-1.5.0-cp36 下载
cu102/torchvision-0.6.0-cp36 下载
两个文件下载在文件夹 C:\Users\ht\Desktop\pytorch
进入pycharm Terminal
C:
cd C:\Users\ht\Desktop\pytorch
dir
pip install torch-1.5.0-cp36-cp36m-win_amd64.whl
结果显示：
Successfully installed future-0.18.2 numpy-1.19.0 torch-1.5.0
pip install torchvision-0.6.0-cp36-cp36m-win_amd64.whl
结果显示：
Successfully installed pillow-7.2.0 torchvision-0.6.0
如果下载速度过慢，添加镜像源
阿里云 http://mirrors.aliyun.com/pypi/simple
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple
豆瓣(douban) http://pypi.douban.com/simple
清华大学 https://pypi.tuna.tsinghua.edu.cn/simple
中国科学技术大学 [http://pypi.mirrors.ustc.edu.cn/simple
pip config set global.index-url https://mirrors.alyun.com/pypi/simple
5.2 配置虚拟环境2
虚拟环境2 python=3.7 pytorch=1.4.0 （CPU）
https://download.pytorch.org/whl/torch_stable.html
Ctrl +F 搜索 cpu/torch-1.4.0%2Bcpu-cp37 下载
cpu/torchvision-0.5.0%2Bcpu-cp37 下载
l两个文件下载在文件夹 C:\Users\ht\Desktop\pytorch
进入pycharm Terminal
deactivate
查看所有虚拟环境
conda info --envs
conda activate pytorch_1.5_cpu
pip install “torch-1.4.0+cpu-cp37-cp37m-win_amd64.whl”
结果显示：
Successfully installed torch-1.4.0+cpu
pip install “torchvision-0.5.0+cpu-cp37-cp37m-win_amd64.whl”
结果显示
Successfully installed numpy-1.19.0 pillow-7.2.0 six-1.15.0 torchvision-0.5.0+cpu
结果显示：
Successfully installed torch-1.4.0+cpu
pip install “torchvision-0.5.0+cpu-cp37-cp37m-win_amd64.whl”
结果显示
Successfully installed numpy-1.19.0 pillow-7.2.0 six-1.15.0 torchvision-0.5.0+cpu
所属网站分类: 技术文章 > 博客
作者：天青色等烟雨
链接： https://www.pythonheidong.com/blog/article/455776/db9defd46c6251155594/
来源： python黑洞网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容：(最多支持255个字符)