win10安装CUDA11.0后再安装GPU版torch的踩坑记录⽬录
———————— 2020. 11. 02 更新 ————————
感谢评论区的 提醒:
安装 cuda11 (坑)
电脑先安装的cuda11.0 , 随便搜索下 cuda download,就会给您跳转到这个页⾯:
然后根据⽹上的教程(为避免报错,不安装 visual studio integration;) ,
顺利地⾛到了安装成功的⼀步,接下来再去搜 torch 安装。
冇得⽀持cuda11.0的!!再根据 到 10.2版本的下载地址,
卸载cuda_V11
先打开【控制⾯板】卸载掉cuda11.0 (根据安装时间和名称含有cuda11的),
只保留 【NVIDIA的图形驱动程序、NVIDIA Physx系统软件,NVIDIA GeForce Experience】,如果你有这3个软件,就别卸载。
安装cuda_V10
下载CUDA10.2 之后重新⾃定义安装(链接同上)。
【注意:为了避免多次安装CUDA 后,安装⽬录竟然消失了,重新安装了三遍,还是消失?的问题,参考】:临时解压⽬录千万不要和cuda的安装路径设置成⼀样的,否则安装结束,会不到安装⽬录的!!
选择【⾃定义安装】 :不要勾选“INVIDIA GeForce ”
安装结束后, 查看环境变量,
添加 %CUDA_path%;%CUDA_PATH%\bin;%CUDA_PATH%\lib\x64;%CUDA_path%\libnvvp; 到path。
pycharm安装教程win10打开cmd验证安装成功
Win + R 打开 cmd
cmd⾥查看CUDA 版本信息 nvcc -V
cmd 查看GPU运⾏时的监测界⾯ nvidia-smi
运⾏⼏个相关的 exe
1. cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite
2. 运⾏
3. 运⾏
安装cuDnn
安装pytorch + torchvision
pip install torch===1.6.0 torchvision===0.7.0 -f /whl/torch_stable.html -i pypi.tuna.tsinghua.edu/simple
验证安装是否成功,打开pycharm 的python console
import torch
print(torch.__version__)
import torchvision
print(torchvision.__version__)
OK,完啦。
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198,我们将在24小时内删除。
发表评论