pytorch安装和测试
文章目录
- 一、显卡驱动安装
- 二、安装anaconda
- 二、cuda安装
- 三、查看path是否有添加成果
- 查看cuda版本
- 四、安装pytorch
- 更新pip
- 安装pytorch
- 五、安装pycharm
- 六、查看pytorch 和 gpu是否可以使用
- 总结
一、显卡驱动安装
搜索适合你电脑的显卡驱动,安装studio版本。
二、安装anaconda
具体安装步骤参考其他教程。
二、cuda安装
选择你的系统支持的cuda。
三、查看path是否有添加成果
进入 :我的电脑-属性-高级系统设置
进入环境变量。
可以安装多个cuda版本。
查看cuda版本
nvcc -V
四、安装pytorch
更新pip
安装pytorch
进入pytorch官网 https://pytorch.org/
选择适合系统的pyrorch版本。
五、安装pycharm
请查看其他教程。
配置pycharm编译环境和配套的包。
六、查看pytorch 和 gpu是否可以使用
import torch
print(torch.__version__)
print('g:',torch.cuda.is_available())
成功了 !!!