Anaconda3+PyTorch安装教程⽂章⽬录
⼀、安装前准备
电脑需要有NVIDA的独⽴显卡;
电脑要有NVIDA显卡驱动程序(NVIDA控制⾯板)
database system微软官⽹下载安装 Microsoft Visual Studio(申请⼀个微软账号)
官⽹下载安装 CUDA Toolkit
检验CUDA安装是否成功pycharm python安装教程
CUDA安装成功后的使⽤
⼆、前提
CUDA在安装之前,需要看NVIDA的官⽅CUDA安装⽂档。
下载安装CUDA需要有NVIDA的显卡、Windows系统、Visual Studio。
三、查看电脑显卡驱动信息
⼀般新买的电脑都带有。
四、下载安装 Visual Studio(⾮必要)勾选“C++的桌⾯开发“,然后下载到本地。
textbox加滚动条选择安装位置进⾏安装。
五、NVIDIA CUDA Toolkit 10.2安装
cuda只能运⾏在英伟达的显卡上
第⼀步,系统检查,如果下载的和NVIDIA控制⾯板结果⼀致,则显⽰绿⾊通过提⽰;第⼆步,许可协议,同意并继续;
第三步,⼀般选择⾃定义安装,选择全部即可
第四步,选择安装位置,可以⾃⼰设置,也可以遵循系统设置(装在C盘)
第五步,如果没有安装Visual Studio,需要勾选下⾯⽅框,然后Next。
第六步,查看版本是否安装成功。
nvcc -V                            # 出现下列说明安装成功
nvcc: NVIDIA ® Cuda compiler driver
Copyright © 2005-2019 NVIDIA Corporation
字符串函数pythonBuiit on Wed Oct 23 19:32:27 Pacific_Daylight_Time 2019
Cuda compilation tools, release 10. 2, v10.2. 89
如果安装不成功,则需要配置环境变量。
电脑右键属性—>⾼级系统设置—>环境变量(系统变量path)
六、创建PyTorch环境
6.1 ⽅法⼀:
常见的五种幂函数图像卸载原来的pytorch和torchvision,之前没有安装则忽略这⼀步。
打开Anaconda Prompt:
①创建PyTorch环境
conda create -n pytorch python=3.7
②激活PyTorch环境
conda activate pytorch            # pytorch为环境名
(pytorch)C:\user>conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch ④查看是否安装成功:
nvcc -V
⑤退出:
deactivate pytorch
⑥删除虚拟环境
conda remove -n 环境名称 --all
6.2 ⽅法⼆(推荐)
直接以管理员⾝份运⾏cmd,然后在串⼝输⼊:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
七、pycharm中添加conda环境(不选择创建虚拟环境)
在pycharm中新建⼀个⼯程(new project)—>选择已经存在的环境解释器(Existing interpreter)—>选择(Conda Environment)—>选择Anaconda安装的python解释器(⽂件)—>勾选Make available to all projects。
说明:这种情况⽅法简单,但是会造成环境污染。⾃⼰电脑中包会⾮常多,不便于后期管理。
⼋、测试pytorch是否安装成功
easyui treegrid动态加载
import torch
print(torch.__version__)# 打印pytorch的版本
print("gpu:",torch.cuda.is_available())# 查看torch是否⽀持gpu
E:\A_code\1\venv\ E:/A_code/1/test.py
1.8.1
gpu: True
Process finished with exit code 0

版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198,我们将在24小时内删除。