小白安装python+cuda+pytorh(gpu版本)

首先说一下,安装前你需要自行安装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
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值