conda create -n pytorch3 python=版本
例如:创建一个叫pytorch2的虚拟环境,python版本为3.7:
conda create -n pytorch2 python=3.7
注意环境名大小写不区分
activate 环境名 #windows系统可以直接用activate
pip install ipykernel
python -m ipykernel install --user --name 环境名
python -m ipykernel install --user --name pytorch2
conda deactivate
jupyter notebook
步骤:
桌面->右击鼠标->NVIDIA控制面板->帮助->系统信息->组件
在NVIDIA控制面板中点击菜单栏中的小房子图标,进入主界面,可以看到GPU版本与驱动版本。如下图所示。
显示GPU当前的状态,在cmd内输入以下代码:
nvidia-smi
去官网:Start Locally | PyTorch
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-9jabk9A4-1640960907854)(C:\Users\Lenovo\AppData\Roaming\Typora\typora-user-images\image-20211217160546507.png)]
一定要都匹配上
下载完成点击安装
右键启动/左上角启动服务:Device Install Service、Device Setup Manager、Windows Installer然后再开始安装。
打开下图的Driver components->Display Driver,看自己的当前版本,若大于等于新版本将 √ 去掉
进行下一步,选择安装位置,一定选择默认位置!!!如果有人尝试下载到别的路径,那么过一会你就会发现CUDA在你的下载路径下消失了!回收站里也没有!amazing!
选择对应cuda版本,torch版本和python版本
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-ffzCGYR6-1640960907856)(C:\Users\Lenovo\AppData\Roaming\Typora\typora-user-images\image-20211217161850047.png)]
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-UKw8QaAO-1640960907857)(C:\Users\Lenovo\AppData\Roaming\Typora\typora-user-images\image-20211217161540713.png)]
我的是:C:\Users\Lenovo>
activate 环境名
pip install xxxx.whl
或者:最后将下载的whl文件随便存放在一个地方,然后打开cmd,输入pip install 路径\文件名.whl
,就开始离线安装了,基本上不会有什么问题。(不过首先得安装好wheel,pip install wheel
)
5.进入python测试
torch.version
torch.cuda.is_available()
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-VpvbP3AJ-1640960907858)(C:\Users\Lenovo\AppData\Roaming\Typora\typora-user-images\image-20211217162211254.png)]
选择对应的torchtext的whl文件下载
activate 环境名
pip install xxxx.whl
或者:最后将下载的whl文件随便存放在一个地方,然后打开cmd,输入pip install 路径\文件名.whl
,就开始离线安装了,基本上不会有什么问题。(不过首先得安装好wheel,pip install wheel
)
conda install matplotlib