小白安装python+cuda+pytorh（gpu版本）

2023-11-08

首先说一下，安装前你需要自行安装anaconda和pycharm环境。

（推荐使用清华镜像安装anaconda https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/）

第1步cuda下载

CUDA下载地址：cuda连接
下载时注意查看自己电脑显卡型号
在这里插入图片描述

打开cuda_10.0.130_411.31_win10.exe安装包
（第一次会提示临时解压路径，完成安装会自动删除，默认/自定义都可以，但不要和安装路径重复，否则会安装失败。一路傻瓜式安装就可以解决。）
安装测试：打开cmd输入nvcc -V 输出以下内容表示CUDA安装成功
在这里插入图片描述

第2步cudnn

cuDNN下载地址：cudnn

注意下载cudnn时要对应你自己刚才下载cuda型号
特别注意 cuDNN下载完成后，解压，将解压后的三个文件夹复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1，每个人地址不一样，不要弄错，就是你刚才安装时的地址。
在这里插入图片描述

第三步

·，https://download.pytorch.org/whl/torch_stable.html
选择自己型号的cuda安装包（是两个，cu就是gpu，torch和version的都要下载，型号要统一）

本人下载的是，要对应好型号，千万注意：
torchvision-0.6.0-cp37-cp37m-win_amd64.whl
torch-1.6.0-cp37-cp37m-win_amd64.whl

第四步

先在anaconda下创建一个gpu的虚拟环境环境
在这里插入图片描述
**open terminal打开终端，**这样省的切换环境时产生错误，说白了就是懒！！

conda create -n paddle python=3.7

说明：-n是代表创建的环境名称 python=3.7表示该环境使用3.7版本python，如果有了可以省略

更新下conda：

conda update --all

使用

pip install torchvision-0.6.0-cp37-cp37m-win_amd64.whl
pip install torch-1.6.0-cp37-cp37m-win_amd64.whl

安装刚才下载好的安装包（每个人不一样）

第五步

在pycharm里添加刚才创建的torch_gpu.
在这里插入图片描述

最后附加各种库国内镜像及使用方法

豆瓣 https://pypi.doubanio.com/simple/
网易 https://mirrors.163.com/pypi/simple/
阿里云 https://mirrors.aliyun.com/pypi/simple/
腾讯云 https://mirrors.cloud.tencent.com/pypi/simple
清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/

# pip install visdom scikit-image tqdm fire ipdb pprint matplotlib torchnet
pip install torchnet -i https://mirrors.aliyun.com/pypi/simple/  --trusted-host pypi.douban.com
pip install visdom -i https://mirrors.aliyun.com/pypi/simple/  --trusted-host pypi.douban.com
pip install scikit-image -i https://mirrors.aliyun.com/pypi/simple/  --trusted-host pypi.douban.com
pip install tqdm -i https://mirrors.aliyun.com/pypi/simple/  --trusted-host pypi.douban.com
pip install fire -i https://mirrors.aliyun.com/pypi/simple/  --trusted-host pypi.douban.com
pip install ipdb -i https://mirrors.aliyun.com/pypi/simple/  --trusted-host pypi.douban.com
pip install pprint -i https://mirrors.aliyun.com/pypi/simple/  --trusted-host pypi.douban.com
pip install matplotlib -i https://mirrors.aliyun.com/pypi/simple/  --trusted-host pypi.douban.com
pip install torchnet -i https://mirrors.aliyun.com/pypi/simple/  --trusted-host pypi.douban.com
pip install numpy -i https://mirrors.aliyun.com/pypi/simple/  --trusted-host pypi.douban.com

本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)