pytorch与Golab

  • 指定显卡跑程序
1  export CUDA_VISIBLE_DEVICES=0

2 在生成网络对象之前:

torch.cuda.set_device(0)

3 初始化模型时

net = Net.cuda(0)

  • pytorch升级
    本文中pytorch由0.4.0升级到1.0.1,再升级至1.1.0(也可以直接升级至1.1.0)
  1. 首先用如下命令查看升级前的pytorch版本,返回版本号是0.4.0
python -c "import torch; print(torch.__version__)"
  1. 输入如下命令 conda install pytorch torchvision -c pytorch进行升级
conda install pytorch torchvision -c pytorc

这个是免费GPU

如果使用pytorch的话就用下面这句话
pip install torch torchvision

  • . 这是一个使用教程:

https://www.jianshu.com/p/a42d69568966

  • 这是网址:

https://colab.research.google.com/notebooks/intro.ipynb#scrollTo=e-Ggopq-pjLW

  • 挂载云盘命令:
from google.colab import drive
drive.mount('/content/gdrive')
  • 切换文件夹命令:
import os
os.chdir("/content/drive/My Drive/###r")
  • linux删除文件:
rm -rf 文件夹
执行linux命令要用“!”
  • 不断线助手
https://blog.csdn.net/DeepFaceLabs/article/details/103889916
function ClickConnect(){
console.log("Working"); 
document.querySelector("colab-toolbar-button#connect").click() 
}
setInterval(ClickConnect,60000)

打开你的浏览器,执行F12或者Ctrl + Shift + i , 将上面代码复制粘贴到Console框里按回车即可。

你可能感兴趣的:(工具使用)