首先说一下,安装前你需要自行安装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